Description
Input
Output
Sample Input
aaa abab
Sample Output
6 6求所有后缀和原串的最长公共前缀和。后缀数组求出h,得到原串的rank 左右扫一遍,后RMQ直接搞。#include<cstring> #include<cstdio> #include<iostream> #define FOR(i,a,b) for(int i=a;i<=b;++i) #define clr(f,z) memset(f,z,sizeof(f)) #define ll(x) (1<<x) using namespace std; const int msize=1e5+9; const int sig=257; class SUFFIX_ARRAY { public: int ran[msize],t1[msize],c[msize],sa[msize]; int n; int idx(char x) { if(x=='\0')return 0; return x-'a'+1; } bool cmp(int*r,int i,int k) { return r[ sa[i] ]==r[ sa[i-1] ]&&r[ sa[i]+k ]==r[ sa[i-1]+k ]; } void build_SA(char*s,int m) { int*wx=t1,*wy=ran; n=strlen(s)+1; FOR(i,0,m-1)c[i]=0; FOR(i,0,n-1)c[ wx[i]=idx(s[i]) ]++; FOR(i,1,m-1)c[i]+=c[i-1]; for(int i=n-1;i>=0;--i)sa[ --c[ wx[i] ] ]=i; for(int k=1;k<=n;k<<=1) { int p=0; FOR(i,n-k,n-1)wy[p++]=i;//二关键字排序 FOR(i,0,n-1)if(sa[i]>=k)wy[p++]=sa[i]-k; FOR(i,0,m-1)c[i]=0; FOR(i,0,n-1)++c[ wx[ wy[i] ] ]; FOR(i,1,m-1)c[i]+=c[i-1]; for(int i=n-1;i>=0;--i)sa[ --c[ wx[ wy[i] ] ] ]=wy[i]; swap(wx,wy); wx[ sa[0] ]=0; p=1; FOR(i,1,n-1)wx[ sa[i] ]=cmp(wy,i,k)?p-1:p++; if(p>=n)break; m=p; } --n; } int h[msize]; void get_H(char*s) { int k=0; FOR(i,0,n)ran[ sa[i] ]=i; FOR(i,0,n-1) { if(k)--k; int j=sa[ ran[i]-1 ]; while( s[i+k]==s[j+k] )++k; h[ ran[i] ]=k; } } void debug() { printf("sa="); FOR(i,0,n)printf("%d ",sa[i]);puts(""); printf("rank="); FOR(i,0,n)printf("%d ",ran[i]);puts(""); printf("h="); FOR(i,0,n)printf("%d ",h[i]);puts(""); } int rmq[msize][32],bit[msize]; void initRMQ() { bit[0]=-1; FOR(i,1,n)bit[i]=(i&(i-1))==0?bit[i-1]+1:bit[i-1]; FOR(i,1,n)rmq[i][0]=h[i]; FOR(i,1,bit[n]) for(int j=1;j+ll(i)-1<=n;++j) rmq[j][i]=min(rmq[j][i-1],rmq[j+ll(i-1)][i-1]); } int LCP(int l,int r) { l=ran[l];r=ran[r]; if(l>r)swap(l,r); ++l;//之前一个已经求过了。 int t=bit[r-l+1]; r-=ll(t)-1; return min(rmq[l][t],rmq[r][t]); } }; SUFFIX_ARRAY ty; char s[msize]; int main() { while(~scanf("%s",s)) { ty.build_SA(s,257); ty.get_H(s); ty.initRMQ(); int hh=1e8; int ans=0; for(int i=1;s[i];++i) { ans+=ty.LCP(0,i); ans%=256; } ans+=strlen(s); ans%=256; printf("%d\n",ans); } }