vijosP1567子串计数

描述
现在有一个字符串，请求出这个字符串不相同的子串个数。
YXY现在不会做，请你来帮忙……

n<=20W

题解：

后缀数组裸题，其实我在练习模板写对了没

代码：

 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #include<iostream>
 #include<vector>
 #include<map>
 #include<set>
 #include<queue>
 #include<string>
 #define inf 1000000000
 #define maxn 200000+5
 #define maxm 500+100
 #define eps 1e-10
 #define ll long long
 #define pa pair<int,int>
 #define for0(i,n) for(int i=0;i<=(n);i++)
 #define for1(i,n) for(int i=1;i<=(n);i++)
 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
 #define mod 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}
     return x*f;
 }
 int n,c[maxn],sa[maxn],rk[maxn],h[maxn],t[maxn],t2[maxn];
 char s[maxn];
 void build(int m)
 {
     int *x=t,*y=t2;
     for0(i,m)c[i]=;
     for0(i,n)c[x[i]=s[i]]++;
     for1(i,m)c[i]+=c[i-];
     for3(i,n,)sa[--c[x[i]]]=i;
     for(int k=;k<=n+;k<<=)
     {
         int p=;
         for2(i,n-k+,n)y[p++]=i;
         for0(i,n)if(sa[i]>=k)y[p++]=sa[i]-k;
         for0(i,m)c[i]=;
         for0(i,n)c[x[y[i]]]++;
         for1(i,m)c[i]+=c[i-];
         for3(i,n,)sa[--c[x[y[i]]]]=y[i];
         swap(x,y);
         p=;x[sa[]]=;
         for1(i,n)x[sa[i]]=y[sa[i-]]==y[sa[i]]&&y[sa[i-]+k]==y[sa[i]+k]?p-:p++;
         if(p>=n+)break;
         m=p-;
     }
 }
 void calch()
 {
     for1(i,n)rk[sa[i]]=i;
     for(int i=,k=,j;i<n;h[rk[i++]]=k)
      for(k?k--:,j=sa[rk[i]-];s[i+k]==s[j+k];k++);
 }
 int main()
 {
     freopen("input.txt","r",stdin);
     freopen("output.txt","w",stdout);
     n=read();int len=;
     while(len<n)
     {
         scanf("%s",s+len);
         len=strlen(s);
     }
     s[n]=' ';
     build();
     calch();
     ll ans=;
     for1(i,n)ans+=(ll)(n-sa[i]-h[i]);
     cout<<ans<<endl;
     return ;
 }

这次的模板比上次好多了。。。