bzoj 4566 找相同字符 —— 廣義後綴自動機
阿新 • • 發佈:2019-01-19
ont -- http right col print typedef class void
題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4566
建出兩個串的廣義後綴自動機;
統計每個點在兩個串中出現次數的子樹和,其實就是在兩個串中的 right 集合大小;
然後統計答案乘起來即可。
代碼如下:
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long ll; int const xn=8e5+5; int n,m,cnt=1,fa[xn],go[xn][30],l[xn],tax[xn],q[xn],d1[xn],d2[xn];char dc[xn]; int work(int p,int w) { int nq=++cnt,q=go[p][w]; l[nq]=l[p]+1; fa[nq]=fa[q]; fa[q]=nq; memcpy(go[nq],go[q],sizeof go[q]); for(;p&&go[p][w]==q;p=fa[p])go[p][w]=nq; return nq; } int ext(int p,int w) { if(go[p][w]) { int q=go[p][w]; if(l[q]==l[p]+1)returnq; return work(p,w); } int np=++cnt; l[np]=l[p]+1; for(;p&&!go[p][w];p=fa[p])go[p][w]=np; if(!p)fa[np]=1; else { int q=go[p][w]; if(l[q]==l[p]+1)fa[np]=q; else fa[np]=work(p,w); } return np; } void rsort() { for(int i=1;i<=cnt;i++)tax[l[i]]++;for(int i=1;i<=cnt;i++)tax[i]+=tax[i-1]; for(int i=cnt;i;i--)q[tax[l[i]]--]=i; } int main() { scanf("%s",dc); n=strlen(dc); for(int lst=1,i=0;i<n;i++)lst=ext(lst,dc[i]-‘a‘+1),d1[lst]++; scanf("%s",dc); m=strlen(dc); for(int lst=1,i=0;i<n;i++)lst=ext(lst,dc[i]-‘a‘+1),d2[lst]++; rsort(); for(int i=cnt,x;i;i--)d1[fa[x=q[i]]]+=d1[x],d2[fa[x]]+=d2[x]; ll ans=0; for(int i=1;i<=cnt;i++)ans+=(ll)(l[i]-l[fa[i]])*d1[i]*d2[i]; printf("%lld\n",ans); return 0; }
bzoj 4566 找相同字符 —— 廣義後綴自動機