bzoj 2528: [Poi2011]Periodicity【kmp+構造】
阿新 • • 發佈:2018-08-20
get() uri [1] swap char lse pri iostream ring
神仙構造,做不來做不來
詳見:http://vfleaking.blog.163.com/blog/static/174807634201329104716122/
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int N=200005; int T,n,ne[N],p[N],cnt,a[N],I,J; char s[N]; void get() { while(1) { if(J==0||a[I]==a[J]) { I++;J++; ne[I]=J; break; } else J=ne[J]; } } int main() { scanf("%d",&T); while(T--) { scanf("%s",s+1); n=strlen(s+1); int i=1,j=0; ne[1]=0; while(i<=n) { if(j==0||s[i]==s[j]) { i++;j++; ne[i]=j; } else j=ne[j]; } i=n+1; cnt=0; while(i!=1) { cnt++; p[cnt]=i-1; i=ne[i]; } for(int i=1;i<=cnt/2;i++) swap(p[i],p[cnt+1-i]); I=2;J=1; if(p[1]==1) a[1]=0; else { a[1]=0; for(int i=2;i<p[1];i++) a[i]=0,get(); a[p[1]]=1,get(); } for(int i=2;i<=cnt;i++) { if(p[i]<=2*p[i-1]) { for(int j=p[i-1]+1;j<=p[i];j++) a[j]=a[j-p[i]+p[i-1]],get(); } else { for(int j=p[i-1]+1;j<p[i]-p[i-1];j++) a[j]=0,get(); int ii=I,jj=J; a[p[i]-p[i-1]]=0;get(); if(ne[I]!=1&&(p[i]-p[i-1])%(I-ne[I])==0) { I=ii;J=jj; a[p[i]-p[i-1]]=1; get(); } for(int j=p[i]-p[i-1]+1;j<=p[i];j++) a[j]=a[j-p[i]+p[i-1]],get(); } } for(i=1;i<=n;i++) printf("%d",a[i]); puts(""); } return 0; }
bzoj 2528: [Poi2011]Periodicity【kmp+構造】