BZOJ[3992][SDOI2015]序列统计 生成函数+NTT

首先了解一下指标
看我瞎bb也可以
因为原根\(g\)满足\(g^i,g^j(i,j\in (1,MOD-1),i\neq j)\)互不相同
则可以给每个数\(i\)定义一个指标\(ind_i\)表示模意义下的\(\log_g i\),并且在区间\([1,\varphi(MOD)]\)中是互不相同的
和\(log\)类似,指标也满足\(ind_{i*j}\equiv ind_i+ind_j\)就可以把乘法弄成加法了
题目要求的\((a×b)mod M=x\)等价于\((ind[a]+ind[b])mod φ(M)=ind[x]\)
所以弄个生成函数再NTT,因为有取模,所以要在每次NTT后将\(\varphi(m)\)项后面的系数挪到前面,剩下的快速幂解决就好了

代码如下:

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#include<cmath>
#define MOD 1004535809
#define N 50020
#define int long long
using namespace std;
const bool DFT=false,IDFT=true;
const int G=3;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<1)+(x<<3)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
typedef long long LL;
void exgcd(LL a,LL b,LL &x,LL &y){
    if(!b){
        x=1;y=0;
        return;
    }
    exgcd(b,a%b,y,x);
    y=y-a/b*x;
    return;
}
inline int GetNi(int k){
    LL x,y;
    exgcd(k,MOD,x,y);
    return (x+MOD)%MOD;
}
inline int qpow(LL x,LL k,LL p){
    LL sum;
    for(sum=1;k;k>>=1,x=x*x%p)
        if(k&1) sum=sum*x%p;
    return sum;
}
bool b[N];
int ind[N],pos[N],sum[N],p[N],q[N],a[N];
int n,m,x,s,t,g,len,inv_len,inv_g;
inline void NTT(int a[],bool mode){
    for(int i=0;i<len;i++)
        if(i<pos[i])
            swap(a[i],a[pos[i]]);
    int g=mode?inv_g:G;
    for(int i=2,mid=1;i<=len;i<<=1,mid<<=1){
        int wm=qpow(g,(MOD-1)/i,MOD);
        for(int j=0;j<len;j+=i){
            for(int k=j,w=1;k<mid+j;k++,w=(LL)w*wm%MOD){
                int l=a[k],r=(LL)w*a[k+mid]%MOD;
                a[k]=(l+r)%MOD;a[k+mid]=(l-r+MOD)%MOD;
            }
        }
    }
    if(mode==IDFT)
        for(int i=0;i<len;i++)
            a[i]=a[i]*inv_len%MOD;
    return;
}
inline void Mul(int a[],int b[],int c[]){
    for(int i=0;i<len;i++)
        p[i]=a[i],q[i]=b[i];
    NTT(p,DFT);NTT(q,DFT);
    for(int i=0;i<len;i++)
        c[i]=(LL)p[i]*q[i]%MOD;
    NTT(c,IDFT);
    for(int i=m-1;i<=m*2-4;i++)
        c[i-m+1]=(c[i-m+1]+c[i])%MOD,c[i]=0;
    return;
}
bool check(int x,int k) {
    for(int i=1;i<k;i++) b[i]=false;
    for(int i=1;i<k;i++){
        int t=qpow(x,i,k);
        if(b[t]) return false;
        b[t]=true;
    }
    return true;
}
inline int GetG(int k){
    int x=1;
    while(!check(x,k)) x++;
    return x;
}
main(){
    inv_g=GetNi(G);
    n=read();m=read();x=read();s=read();
    g=GetG(m);
    for(int i=0,now=1;i<m-1;i++,(now*=g)%=m) ind[now]=i;
    for(int i=1;i<=s;i++){
        t=read()%m;
        if(t) a[ind[t]]=1;
    }
    for(len=1;len<m<<1;len<<=1);
    for(int i=0;i<len;i++){
        pos[i]=pos[i>>1]>>1;
        if(i&1) pos[i]|=len>>1;
    }
    inv_len=GetNi(len);
    sum[0]=1;
    while(n){
        if(n&1) Mul(sum,a,sum);
        Mul(a,a,a);
        n>>=1;
    }
    printf("%d",sum[ind[x]]);
return 0;
}

原文地址:https://www.cnblogs.com/Duan2baka/p/8674399.html

时间: 2024-10-11 01:45:49

BZOJ[3992][SDOI2015]序列统计 生成函数+NTT的相关文章

BZOJ 3992: [SDOI2015]序列统计 NTT+快速幂

3992: [SDOI2015]序列统计 Time Limit: 30 Sec  Memory Limit: 128 MBSubmit: 1155  Solved: 532[Submit][Status][Discuss] Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S. 小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中

BZOJ 3992: [SDOI2015]序列统计 [快速数论变换 生成函数 离散对数]

3992: [SDOI2015]序列统计 Time Limit: 30 Sec  Memory Limit: 128 MBSubmit: 1017  Solved: 466[Submit][Status][Discuss] Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S. 小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中

BZOJ 3992: [SDOI2015]序列统计 快速幂+NTT(离散对数下)

3992: [SDOI2015]序列统计 Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S. 小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个.小C认为,两个数列{Ai}和{Bi}不同,当且仅当至少存在一个整数i,满足Ai≠Bi.另外,小C认为这个问题的答案可能很大,因

bzoj 3992: [SDOI2015]序列统计

膝盖++,IQ-- SD总是酱紫.....吐槽+++++++ 这个乘积的形式是可以用他的原根表示成加法的!!神奇啊!!! 然后加法就很棒棒了,我们可以用生成函数这个东西来计算一下了. 然后NTT就好了!! 还有这里有一个像快速幂的东西,而且把大于模数的东西搞小,是循环卷积的形式吗??好神奇啊 原根真的是劲啊 1 #include <cstdio> 2 #include <iostream> 3 #include <cmath> 4 #define LL long lon

BZOJ 3992 Sdoi2015 序列统计 快速数论变换

题目大意:给定n(n<=10^9),质数m(3<=m<=8000),1<=x=m,以及一个[0,m-1]区间内的集合S,求有多少长度为n的数列满足每个元素都属于集合S且所有元素的乘积mod m后=x 求原根,对S集合内每个元素取指标,然后搞出生成函数f(x) 那么答案就是(f(x))^n (mod x^(m-1),mod 1004535809) 上NTT用多项式快速幂搞一搞就好了 #include <cstdio> #include <cstring> #i

[bzoj3992][SDOI2015]序列统计——离散对数+NTT

题目大意: 给定一个数字不超过\(m\)的集合\(S\),用\(S\)中的数生成一个长度为\(n\)的序列,求所有序列中的元素乘积模\(m\)等于\(x\)的序列的个数. 思路: 考虑最朴素的\(DP\),设\(f_{i,j}\)为选了\(i\)个数,乘积模\(m\)余\(j\)的方案数,直接转移的时间复杂度是\(O(nm^2)\)的. 不难发现每次转移的过程是相同的,矩阵加速显然不太可行,考虑将乘法形式的转移变成加法形式的转移,这样每次转移即可用NTT优化. 这里需要用到一个叫做离散对数的东西

【BZOJ3992】[SDOI2015]序列统计 NTT+多项式快速幂

[BZOJ3992][SDOI2015]序列统计 Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S. 小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个.小C认为,两个数列{Ai}和{Bi}不同,当且仅当至少存在一个整数i,满足Ai≠Bi.另外,小C认为这个问题的答案可能

BZOJ3992 [SDOI2015]序列统计 【生成函数 + 多项式快速幂】

题目 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数 列,数列中的每个数都属于集合S.小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助: 给定整数x,求所有可以生成出的,且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个.小C认为 ,两个数列{Ai}和{Bi}不同,当且仅当至少存在一个整数i,满足Ai≠Bi.另外,小C认为这个问题的答案可能很大 ,因此他只需要你帮助他求出答案mod 1004535809

bzoj3992 [SDOI2015]序列统计

Description 小C有一个集合S,里面的元素都是小于M的非负整数.他用程序编写了一个数列生成器,可以生成一个长度为N的数列,数列中的每个数都属于集合S. 小C用这个生成器生成了许多这样的数列.但是小C有一个问题需要你的帮助:给定整数x,求所有可以生成出的,且满足数列中所有数的乘积mod M的值等于x的不同的数列的有多少个.小C认为,两个数列{Ai}和{Bi}不同,当且仅当至少存在一个整数i,满足Ai≠Bi.另外,小C认为这个问题 的答案可能很大,因此他只需要你帮助他求出答案mod 100