BZOJ3160【万径人踪灭】 【FFT】

。。恩 打了四五遍 不会也背出来了。。

BZOJ3160 【听说时限紧?转C++的优势么?】

上AC代码 fft

 1 /*Problem: 3160
 2     User: cyz666
 3     Language: C++
 4     Result: Accepted
 5     Time:1992 ms
 6     Memory:18492 kb
 7 ****************************************************************/
 8  
 9 #include <bits/stdc++.h>
10 #define LL long long
11 const LL mo=1000000007;
12 const double pi=acos(-1.0);
13 using namespace std;
14 struct cp{
15     double x,y;
16     cp(double a=0,double b=0):x(a),y(b){}
17     //cp(double a=0,double b=0){x=a,y=b;}
18 }a[400005],b[400005],W[200005];
19 int h[400005],rev[400005],f[200005],c[200005],n,x,m; 
20 char s; LL ans;
21 cp operator +(cp a,cp b){
22     cp c(a.x+b.x,a.y+b.y);
23     return c;
24 }
25 cp operator *(cp a,cp b){
26     cp c(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
27     return c;
28 }
29 cp operator -(cp a,cp b){
30     return (cp){a.x-b.x,a.y-b.y};
31 }
32 void fft(cp a[],int n,int d=1){
33     if (d<0) reverse(a+1,a+n);
34     for (int i=0;i<n;++i)
35         if (i<rev[i]) swap(a[i],a[rev[i]]);
36     for (int i=0;i<n/2;++i)
37         W[i]=cp(cos(2*pi/n*i),sin(2*pi/n*i));
38     for (int m=2,k=1;m<=n;k=m,m<<=1)
39         for (int i=0;i<n;i+=m)
40         for (int j=i;j<i+k;++j){
41             cp u=a[j],v=a[j+k]*W[n/m*(j-i)];
42             a[j]=u+v,a[j+k]=u-v;
43         }
44     if (d<0) for (int i=0;i<n;++i) a[i].x/=n;
45 }
46 int main(){
47     while (scanf("%c",&s)&&isalpha(s))
48         s=='a'?c[++++n]=1:c[++++n]=2;
49     c[0]=-1; c[n+2]=-2;
50     for (int i=1;i<=n;++i){
51         if (x+f[x]>i) f[i]=min(f[x+x-i],x+f[x]-i);
52         while (c[i+f[i]]==c[i-f[i]]) ++f[i]; --f[i];
53         if (i+f[i]>x+f[x]) x=i;
54         ans+=f[i]/2; if (ans>=mo) ans-=mo;
55     }
56     m=ceil((log(n)/log(2))); m=1<<m;
57     for (int i=1;i<m;++i)rev[i]=(rev[i>>1]>>1)+(m>>1)*(i&1);
58     x=0; for (int i=2;i<=n;++++i) c[++x]=c[i]; n=x;
59     for (int i=1;i<=n;++i) c[i]==1?a[i].x=1:b[i].x=1;
60     fft(a,m);  fft(b,m);
61     for (int i=0;i<m;++i) a[i]=a[i]*a[i],b[i]=b[i]*b[i];
62     fft(a,m,-1); fft(b,m,-1);
63     h[0]=1; for (int i=1;i<=n;++i)h[i]=(h[i-1]+h[i-1])%mo;
64     int x;  ans=mo-ans; 
65     for (int i=1;i<n+n;++i){
66         x=round(a[i].x+b[i].x); 
67         ans+=h[x+1>>1]-1; if (ans>=mo) ans-=mo;
68     }
69     ans=(ans-n+1+mo)%mo;
70     printf("%lld",ans);
71     return 0;
72 }
KriTo

  

然后 顺手写了个ntt板子扔这(没用题目测过对不对)

mo的原根g的定义:g^1,g^2,...,g^(mo-1) 在%mo意义下各不相同。

 1 #include <bits/stdc++.h>
 2 #define LL long long
 3 using namespace std;
 4 int g1,g2,N,n,k,x,W[1050000],a[1050000],b[1050000],rev[1050000],mo1,mo2;
 5 LL po(LL x,LL y,LL mo){
 6     LL z=1; if (y<0) y=y%(mo-1)+mo-1;
 7     for (;y;y>>=1,x=x*x%mo)
 8     if (y&1) z=z*x%mo;
 9     return z;
10 }
11 int getg(LL mo){
12     LL y=mo-1,x=floor(sqrt(y)); int fl; 
13     for (int g=2;;++g){
14         fl=1;
15         for (int i=2;i<=x;++i) if (!(y%i))
16         if (po(g,i,mo)==1||po(g,y/i,mo)==1) {fl=0;break;}
17         if (fl) return g;
18     }
19 }
20 void ntt(int a[],int n,int d,int mo,int G){    //n整除(mo-1)
21     if (d<0) reverse(a+1,a+n);
22     for (int i=0;i<n;++i)
23         if (i<rev[i]) swap(a[i],a[rev[i]]);
24     W[0]=1; LL X=po(G,(mo-1)/n,mo);
25     for (int i=1;i<n/2;++i) W[i]=X*W[i-1]%mo;
26     for (int m=2,k=1;m<=n;k=m,m<<=1)
27         for (int i=0;i<n;i+=m)
28         for (int j=i;j<i+k;++j){
29             int u=a[j],v=1ll*a[j+k]*W[n/m*(j-i)]%mo;
30             a[j]=u+v>=mo?u+v-mo:u+v;
31             a[j+k]=u-v<0?u-v+mo:u-v;
32         }
33     if (d<0){
34         X=po(n,mo-2,mo);
35         for (int i=0;i<n;++i) a[i]=X*a[i]%mo;
36     }
37 }
38 int main(){
39     mo1=998244353; mo2=1004535809;
40     g1=getg(mo1); g2=getg(mo2);
41     scanf("%d%d",&n,&k); W[0]=1;
42     for (int i=1;i<=n;++i)
43         scanf("%d",&x),a[x]=1,b[x]=1;
44     N=1048576;
45     for (int i=0;i<N;++i) rev[i]=rev[i>>1]>>1|(i&1?N>>1:0);
46     ntt(a,N,1,mo1,g1); ntt(b,N,1,mo2,g2);
47     for (int i=0;i<N;++i)
48         a[i]=po(a[i],k,mo1),b[i]=po(b[i],k,mo2);
49     ntt(a,N,-1,mo1,g1); ntt(b,N,-1,mo2,g2);
50     for (int i=0;i<N;++i){
51         if (a[i]<0) a[i]+=mo1;
52         if (b[i]<0) b[i]+=mo2;
53     }
54     for (int i=0;i<N;++i) if (a[i]||b[i]) printf("%d ",i); puts("");
55     return 0;
56 }
AsuNa

 

若多项式相乘的最高次为n, 则FFT,NTT的数组大小要开到>n的最小的2的幂。 即 1<<(int)ceil(log(n+1)/log(2))

给几个模数:

1004535809=479*2^21+1,   原根=3

998244353=7*17*2^23+1 ,原根=3

469762049=7*2^26+1      ,原根=3

posted @ 2017-02-09 16:45  cyz666  阅读(200)  评论(0编辑  收藏  举报