字符串（后缀数组）：HAOI2016 找相同子串

[HAOI2016]找相同子串

【题目描述】

给定两个字符串，求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两个子串中有一个位置不同。

【输入格式】

两行，两个字符串s1,s2，长度分别为n1,n2。

【输出格式】

输出一个整数表示答案。

【样例输入】

aabb
bbaa

【样例输出】

10

【数据范围】

对于20%的数据，满足1≤n1,n2≤500；

对于40%的数据，满足1≤n1,n2≤5000；

对于100%的数据，满足1≤n1,n2≤200000，字符串中只有小写字母。

　　和POJ 3415几乎一样。

 #include <iostream>
 #include <cstring>
 #include <cstdio>
 using namespace std;
 const int N=;
 char s[N];int len,d;
 int r[N],Wa[N],Wb[N],sa[N],rk[N];
 int Ws[N],Wv[N],lcp[N];

 bool cmp(int *p,int i,int j,int l){
     return p[i]==p[j]&&p[i+l]==p[j+l];
 }

 void DA(int n,int m){
     int i,j,p,*x=Wa,*y=Wb;
     for(i=;i<m;i++)Ws[i]=;
     for(i=;i<n;i++)++Ws[x[i]=r[i]];
     for(i=;i<m;i++)Ws[i]+=Ws[i-];
     for(i=n-;i>=;i--)sa[--Ws[x[i]]]=i;
     for(j=,p=;p<n;j<<=,m=p){
         for(p=,i=n-j;i<n;i++)y[p++]=i;
         for(i=;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
         for(i=;i<m;i++)Ws[i]=;
         for(i=;i<n;i++)++Ws[Wv[i]=x[y[i]]];
         for(i=;i<m;i++)Ws[i]+=Ws[i-];
         for(i=n-;i>=;i--)sa[--Ws[Wv[i]]]=y[i];
         for(swap(x,y),p=,x[sa[]]=,i=;i<n;i++)
             x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
     }
 }

 void LCP(int n){
     int i,j,k=;
     for(i=;i<=n;i++)rk[sa[i]]=i;
     for(i=;i<n;lcp[rk[i++]]=k)
         for(k?k--:k,j=sa[rk[i]-];r[i+k]==r[j+k];k++);
 }

 int ID(int x){x=sa[x];
     if(x==d)return ;
     if(x<d)return ;
     return ;
 }

 int st[N]={-},top;
 long long sum[N],tot[N],ans;
 int main(){
     freopen("find_2016.in","r",stdin);
     freopen("find_2016.out","w",stdout);
     scanf("%s",s);
     len=d=strlen(s);
     scanf("%s",s+len+);
     s[d]='#';len=strlen(s);
     for(int i=;i<len;i++)r[i]=s[i];
     DA(len+,);LCP(len);
     for(int i=;i<=len;i++){
         if(!ID(i))continue;
         if(ID(i)==)ans+=tot[top];
         if(i!=len){
             st[++top]=lcp[i+];sum[top]=-ID(i);
             tot[top]=st[top]*sum[top]+tot[top-];
             while(st[top]<=st[top-]){
                 sum[top-]+=sum[top];
                 st[top-]=st[top];top--;
                 tot[top]=st[top]*sum[top]+tot[top-];
             }
         }
     }
     top=;
     for(int i=;i<=len;i++){
         if(!ID(i))continue;
         if(ID(i)==)ans+=tot[top];
         if(i!=len){
             st[++top]=lcp[i+];sum[top]=ID(i)-;
             tot[top]=st[top]*sum[top]+tot[top-];
             while(st[top]<=st[top-]){
                 sum[top-]+=sum[top];
                 st[top-]=st[top];top--;
                 tot[top]=st[top]*sum[top]+tot[top-];
             }
         }
     }
     printf("%lld\n",ans);
     return ;
 }