1. 程式人生 > >bzoj 4566 找相同字符 —— 廣義後綴自動機

bzoj 4566 找相同字符 —— 廣義後綴自動機

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)return
q; 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 找相同字符 —— 廣義後綴自動機