【CF827E】Rusty String 調和級數+FFT
阿新 • • 發佈:2018-01-29
esp style light ora () name include {} %d
【CF827E】Rusty String
題意:給你一個01串,其中部分字符是‘?‘,?可以是0或1,求所有可能的d,滿足存在一種可能得到的01串,在向右移動d格後與自己相同。
$n\le 5\times 10^5$
題解:我們先枚舉d,那麽一個串符合條件當且僅當:$\forall i \in [0,d), a_i=a_{i+d}=a_{i+2d}=...$,我們可以用fft來判斷是否相等。具體地,我們令a串代表正串中的V,b串代表反串中的K,求a和b的卷積c,這樣只需要判斷$c_d,c_{2d},c_{3d}...$是否都是0就行了。時間復雜度$O(n \log n)$。
#include <cstdio> #include <cstring> #include <iostream> #include <cmath> #include <algorithm> #define pi acos(-1.0) using namespace std; struct cp { double x,y; cp () {} cp (double a,double b) {x=a,y=b;} cp operator + (const cp &a) const {return cp(x+a.x,y+a.y);} cp operator - (const cp &a) const {return cp(x-a.x,y-a.y);} cp operator * (const cp &a) const {return cp(x*a.x-y*a.y,x*a.y+y*a.x);} }l1[1<<20],l2[1<<20]; int n,len,top; int v[1<<20],sta[500010]; char str[500010]; inline void FFT(cp *a,int f) { int i,j,k,h; for(i=k=0;i<len;i++) { if(i>k) swap(a[i],a[k]); for(j=len>>1;(k^=j)<j;j>>=1); } cp t; for(h=2;h<=len;h<<=1) { cp wn(cos(2*pi*f/h),sin(2*pi*f/h)); for(i=0;i<len;i+=h) { cp w(1,0); for(j=i;j<i+h/2;j++) t=w*a[j+h/2],a[j+h/2]=a[j]-t,a[j]=a[j]+t,w=w*wn; } } } inline void work() { scanf("%d%s",&n,str); for(len=1;len<2*n;len<<=1); int i,j; for(i=0;i<n;i++) l1[i].x=(str[i]==‘V‘),l2[i].x=(str[n-i-1]==‘K‘); FFT(l1,1),FFT(l2,1); for(i=0;i<len;i++) l1[i]=l1[i]*l2[i]; FFT(l1,-1); for(i=0;i<len;i++) v[i]=int(l1[i].x/len+0.5); for(top=0,i=1;i<=n;i++) { for(j=(n-1)%i;j<2*n;j+=i) if(v[j]) break; if(j>=2*n) sta[++top]=i; } printf("%d\n",top); for(i=1;i<=top;i++) printf("%d%c",sta[i],i==top?‘\n‘:‘ ‘); memset(l1,0,sizeof(l1[0])*(len+1)),memset(l2,0,sizeof(l2[0])*(len+1)); } int main() { int T; scanf("%d",&T); while(T--) work(); return 0; }//3 5 V??VK 6 ?????? 4 ?VK?
【CF827E】Rusty String 調和級數+FFT