●BZOJ 3172 [Tjoi2013]單詞
阿新 • • 發佈:2017-12-06
pac scan har ios cst itl cstring std break
題鏈:
http://www.lydsy.com/JudgeOnline/problem.php?id=3172
題解:
把單詞逐個接起來,中間用互不相同的字符連接,並記錄下每個單詞的首字母在串中的位置。
(因為太多,所以轉成int數組,用大數連接)
然後計算出 sa[],rak[],hei[]數組。
並求出 L[i]: 找出排名最小(即為 L[i])的後綴與排名為 i的後綴的 LCP==hei[i]
同理R[i]:找出排名最大(即為 R[i])的後綴與排名為 i的後綴的 LCP==hei[i]
可以用單調棧 O(n)維護。
然後對於每個單詞,只要它在後綴數組中與它前面一個後綴或後面一個後綴的LCP==len,
那麽答案就是 R[]-L[]+1,否則就為 1。
代碼:
#include<cstdio> #include<cstring> #include<iostream> #define MAXN 1005000 #define filein(x) freopen(#x".in","r",stdin); #define fileout(x) freopen(#x".out","w",stdout); using namespace std; char S[MAXN]; int A[MAXN],P[MAXN],sa[MAXN],rak[MAXN],hei[MAXN],L[MAXN],R[MAXN]; void build(int N,int M){ static int cc[MAXN],ta[MAXN],tb[MAXN],*x,*y,h,p; x=ta; y=tb; h=0; for(int i=0;i<M;i++) cc[i]=0; for(int i=0;i<N;i++) cc[x[i]=A[i]]++; for(int i=1;i<M;i++) cc[i]+=cc[i-1]; for(int i=N-1;i>=0;i--) sa[--cc[x[i]]]=i; for(int k=1;p=0,k<N;k<<=1){ for(int i=N-k;i<N;i++) y[p++]=i; for(int i=0;i<N;i++) if(sa[i]>=k) y[p++]=sa[i]-k; for(int i=0;i<M;i++) cc[i]=0; for(int i=0;i<N;i++) cc[x[y[i]]]++; for(int i=1;i<M;i++) cc[i]+=cc[i-1]; for(int i=N-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i]; swap(x,y); y[N]=-1; x[sa[0]]=0; M=1; for(int i=1;i<N;i++) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?M-1:M++; if(M>=N) break; } for(int i=0;i<N;i++) rak[sa[i]]=i; A[N]=-1; for(int i=0,j;i<N;i++){ if(h) h--; if(rak[i]){ j=sa[rak[i]-1]; while(A[i+h]==A[j+h]) h++; } hei[rak[i]]=h; } } void pre(int N){ static int stk[MAXN],stp[MAXN],top; top=0; stp[top]=0; for(int i=0;i<N;i++){ while(top&&stk[top]>=hei[i]) top--; L[i]=stp[top]; top++; stk[top]=hei[i]; stp[top]=i; } top=N; stp[top]=N; for(int i=N-1;i>=0;i--){ while(top<N&&stk[top]>=hei[i]) top++; R[i]=stp[top]-1; top--; stk[top]=hei[i]; stp[top]=i; } } int main() { int len=0,N; scanf("%d",&N); for(int i=1;i<=N;i++){ P[i]=len; scanf("%s",S+len); len=strlen(S); S[len++]=‘&‘; S[len]=0; } P[N+1]=len=strlen(S); for(int i=0,cnt=10;i<len;i++) A[i]=S[i]==‘&‘?MAXN-cnt++:S[i]; /* for(int i=1;i<=N;i++) printf("%d ",P[i]); printf("\n"); for(int i=0;i<len;i++) printf("%c ",S[i]); printf("\n"); for(int i=0;i<len;i++) printf("%d ",A[i]); printf("\n");*/ build(len,MAXN-5); pre(len); for(int i=1,ANS,r,l;i<=N;i++){ r=rak[P[i]]; l=P[i+1]-P[i]-1; if(r==0||hei[r+1]==l) r++; if(hei[r]!=l) ANS=1; else ANS=R[r]-L[r]+1; printf("%d\n",ANS); } return 0; }
●BZOJ 3172 [Tjoi2013]單詞