bzoj 3238: [Ahoi2013]差异

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<cstdlib>
 5 #include<cmath>
 6 #include<queue>
 7 #include<algorithm>
 8 #include<vector>
 9 #define M 2000009
10 #define EPS 1e-10
11 #define MO 19650827
12 #define ll long long
13 using namespace std;
14 ll read()
15 {
16     char ch=getchar();
17     ll x=0,f=1;
18     for(;ch<‘0‘||ch>‘9‘;ch=getchar())
19         if(ch==‘-‘)
20           f=-1;
21     for(;ch>=‘0‘&&ch<=‘9‘;ch=getchar())
22         x=x*10+ch-‘0‘;
23     return x*f;
24 }
25 char ch[M];
26 int K,sa[2][M],rk[2][M],v[M],a[M],p=1,q,n,H[M],t;
27 ll ans,b[M],c[M];
28 void work(int sa[M],int rk[M],int SA[M],int RK[M])
29 {
30   for(int i=1;i<=n;i++)
31     v[rk[sa[i]]]=i;
32   for(int i=n;i;i--)
33     if(sa[i]>K)
34       SA[v[rk[sa[i]-K]]--]=sa[i]-K;
35   for(int i=n-K+1;i<=n;i++)
36     SA[v[rk[i]]--]=i;
37   for(int i=1;i<=n;i++)
38     RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+K]!=rk[SA[i-1]+K]);
39 }
40 int main()
41 {
42    scanf("%s",ch+1);
43    n=strlen(ch+1);
44    for(int i=1;i<=n;i++)
45      a[i]=ch[i]-‘a‘+1;
46    for(int i=1;i<=n;i++)
47      v[a[i]]++;
48    for(int i=1;i<=26;i++)
49      v[i]+=v[i-1];
50    for(int i=1;i<=n;i++)
51      sa[p][v[a[i]]--]=i;
52    for(int i=1;i<=n;i++)
53      rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
54    K=1;
55    for(;K<n;)
56      {
57        work(sa[p],rk[p],sa[q],rk[q]);
58        swap(p,q);
59        K<<=1;
60      }
61    K=0;
62    for(int i=1;i<=n;i++)
63      if(rk[p][i]==1)
64        H[i]=0;
65      else
66        {
67          int j=sa[p][rk[p][i]-1];
68          for(;a[i+K]==a[j+K];K++);
69          H[i]=K;
70          if(K)
71            K--;
72        }
73    ll su=0;
74    for(int i=n-1;i;i--)
75      {
76         ll a1=H[sa[p][i+1]],s=1;
77         for(;b[t]>a1;)
78           {
79             s+=c[t];
80             su-=b[t]*c[t];
81             t--;
82            }
83         su+=a1*s;
84         if(b[t]==a1)
85           c[t]+=s;
86         else 
87           {
88             b[++t]=a1;
89             c[t]=s;
90           }
91         ans+=su;
92      }
93    printf("%lld\n",(ll)(n-1)*n*(n+1)/2-2*ans);
94    return 0;
95 }
96