bzoj 3238: [Ahoi2013]差异

一看字符串 最长公共前缀，用后缀数组+单调栈搞搞就行啦。一定要注意long long 啊

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<queue>
#include<algorithm>
#include<vector>
#define M 2000009
#define EPS 1e-10
#define MO 19650827
#define ll long long
using namespace std;
ll read()
{
    char ch=getchar();
    ll x=0,f=1;
    for(;ch<'0'||ch>'9';ch=getchar())
        if(ch=='-')
          f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())
        x=x*10+ch-'0';
    return x*f;
}
char ch[M];
int K,sa[2][M],rk[2][M],v[M],a[M],p=1,q,n,H[M],t;
ll ans,b[M],c[M];
void work(int sa[M],int rk[M],int SA[M],int RK[M])
{
  for(int i=1;i<=n;i++)
    v[rk[sa[i]]]=i;
  for(int i=n;i;i--)
    if(sa[i]>K)
      SA[v[rk[sa[i]-K]]--]=sa[i]-K;
  for(int i=n-K+1;i<=n;i++)
    SA[v[rk[i]]--]=i;
  for(int i=1;i<=n;i++)
    RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+K]!=rk[SA[i-1]+K]);
}
int main()
{
   scanf("%s",ch+1);
   n=strlen(ch+1);
   for(int i=1;i<=n;i++)
     a[i]=ch[i]-'a'+1;
   for(int i=1;i<=n;i++)
     v[a[i]]++;
   for(int i=1;i<=26;i++)
     v[i]+=v[i-1];
   for(int i=1;i<=n;i++)
     sa[p][v[a[i]]--]=i;
   for(int i=1;i<=n;i++)
     rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
   K=1;
   for(;K<n;)
     {
       work(sa[p],rk[p],sa[q],rk[q]);
       swap(p,q);
       K<<=1;
     }
   K=0;
   for(int i=1;i<=n;i++)
     if(rk[p][i]==1)
       H[i]=0;
     else
       {
         int j=sa[p][rk[p][i]-1];
         for(;a[i+K]==a[j+K];K++);
         H[i]=K;
         if(K)
           K--;
       }
   ll su=0;
   for(int i=n-1;i;i--)
     {
        ll a1=H[sa[p][i+1]],s=1;
        for(;b[t]>a1;)
          {
            s+=c[t];
            su-=b[t]*c[t];
            t--;
           }
        su+=a1*s;
        if(b[t]==a1)
          c[t]+=s;
        else 
          {
            b[++t]=a1;
            c[t]=s;
          }
        ans+=su;
     }
   printf("%lld
",(ll)(n-1)*n*(n+1)/2-2*ans);
   return 0;
}