2377. 「AHOI2013」 差异
给定一个长度为 $n$ 的字符串 $S$ ,令 $T_i$ 表示它从第 $i$ 个字符开始的后缀,求:
$$sum_{1 le i < j le n} operatorname{len}(T_i)+operatorname{len}(T_j)-2operatorname{lcp}(T_i,T_j)$$
sol
建出反串,计算每个点会被当几次lcp(有几对特殊点经过)
我居然忘记ch[i]-'a'了气死
#include<cstdio> #include<iostream> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> #define maxn 1000005 using namespace std; int n,rt,cnt,la,sz[maxn]; long long val[maxn],ans; char ch[maxn]; struct node{ int par,nex[26],Max; }s[maxn]; void ins(int c){ int p=la,np=++cnt;la=np;s[np].Max=s[p].Max+1;sz[np]=1; for(;p&&!s[p].nex[c];p=s[p].par)s[p].nex[c]=np; if(!p)s[np].par=rt; else { int q=s[p].nex[c],nq; if(s[q].Max==s[p].Max+1)s[np].par=q; else { nq=++cnt;s[nq].Max=s[p].Max+1; for(int j=0;j<26;j++)s[nq].nex[j]=s[q].nex[j]; s[nq].par=s[q].par;s[q].par=s[np].par=nq; for(;p&&s[p].nex[c]==q;p=s[p].par)s[p].nex[c]=nq; } } } int tax[maxn],ord[maxn]; void Sort(){ for(int i=1;i<=cnt;i++)tax[s[i].Max]++; for(int i=1;i<=n;i++)tax[i]+=tax[i-1]; for(int i=1;i<=cnt;i++)ord[tax[s[i].Max]--]=i; } int main(){ scanf(" %s",ch);n=strlen(ch); la=rt=cnt=1; for(int i=n-1;i>=0;i--)ins(ch[i]-'a'); Sort(); for(int i=cnt;i>=1;i--){ int x=ord[i]; val[s[x].par]+=1LL*sz[s[x].par]*sz[x]; sz[s[x].par]+=sz[x]; } for(int i=1;i<=n;i++)ans=ans+1LL*i*(n-1); for(int i=1;i<=cnt;i++)ans=ans-val[i]*s[i].Max*2; cout<<ans<<endl; return 0; }