vijosP1567子串计数
现在有一个字符串,请求出这个字符串不相同的子串个数。
输入格式
第一行有一个正整数n,表示字符串的长度。
下面是这个长度为n的字符串,每行80个字符(最后一行可能少于80个)。
输出格式
一个正整数,表示不相同子串的个数
裸后缀数组,以第i个开头的不同的字串个数是n-sa[i]-h[i]
1 #include<iostream> 2 #include<cstdlib> 3 #include<cmath> 4 #include<cstring> 5 #include<cstdio> 6 #include<algorithm> 7 #include<string> 8 #include<map> 9 #include<queue> 10 #include<vector> 11 #include<set> 12 #define inf 1000000000 13 #define maxn 200000+5 14 #define maxm 600 15 #define eps 1e-10 16 #define ll long long 17 #define for0(i,n) for(int i=0;i<=(n);i++) 18 #define for1(i,n) for(int i=1;i<=(n);i++) 19 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 20 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 21 #define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go) 22 using namespace std; 23 int read(){ 24 int x=0,f=1;char ch=getchar(); 25 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 26 while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();} 27 return x*f; 28 } 29 int n,len,a[maxn],sa[maxn],t[maxn],t2[maxn],c[maxn],h[maxn],rk[maxn],b[maxn]; 30 char s[maxn]; 31 void getsa(int m) 32 { 33 int *x=t,*y=t2; 34 for0(i,m)c[i]=0; 35 for0(i,n)c[x[i]=s[i]]++; 36 for1(i,m)c[i]+=c[i-1]; 37 for3(i,n,0)sa[--c[x[i]]]=i; 38 for(int k=1;k<=n+1;k<<=1) 39 { 40 int p=0; 41 for2(i,n-k+1,n)y[p++]=i; 42 for0(i,n)if(sa[i]>=k)y[p++]=sa[i]-k; 43 for0(i,m)c[i]=0; 44 for0(i,n)c[x[y[i]]]++; 45 for1(i,m)c[i]+=c[i-1]; 46 for3(i,n,0)sa[--c[x[y[i]]]]=y[i]; 47 swap(x,y);p=0;x[sa[0]]=0; 48 for1(i,n)x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p; 49 if(p>=n)break; 50 m=p; 51 } 52 for1(i,n)rk[sa[i]]=i; 53 for(int i=0,k=0,j;i<n;h[rk[i++]]=k) 54 for(k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++); 55 } 56 int main(){ 57 //freopen("input.txt","r",stdin); 58 //freopen("output.txt","w",stdout); 59 n=read(); 60 while(len<n){ 61 scanf("%s",s+len); 62 len=strlen(s); 63 } 64 s[n]=' '; 65 getsa(128); 66 ll ans=0; 67 for1(i,n)ans+=(ll)n-sa[i]-h[i]; 68 printf("%lld",ans); 69 return 0; 70 }