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

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

http sca com get con max 集合 const [ ]

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4566

每個後綴結尾處 ct[ ] = 1 ,按拓撲序 dp 一下就能求出 right 集合的大小。自動機上每個點的貢獻就是 ( l [cr]-l [fa] ) * ct[0][cr] * ct[1][cr] , ct[0] 和 ct[1] 表示在兩個字符串裏分別的出現次數。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=2e5+5,M=8e5+5,K=30; int tot=1,go[M][K],fa[M],l[M],ct[2][M],ans[M]; char s[2][N]; int q[M],tx[N],vis[M]; int cz(int p,int w) { int q=go[p][w],nq=++tot;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 ins(int p,int w) { if(go[p][w]) { int q=go[p][w]; if(l[q]==l[p]+1)return q; return cz(p,w); } int np=++tot;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]=cz(p,w); }
return np; } void Rsort(int n) { for(int i=1;i<=tot;i++)tx[l[i]]++; for(int i=1;i<=n;i++)tx[i]+=tx[i-1]; for(int i=1;i<=tot;i++)q[tx[l[i]]--]=i; } int main() { scanf("%s",s[0]);scanf("%s",s[1]); int n[2];n[0]=strlen(s[0]);n[1]=strlen(s[1]); for(int t=0;t<=1;t++) for(int pr=1,i=0;i<n[t];i++)pr=ins(pr,s[t][i]-a),ct[t][pr]=1; Rsort(max(n[0],n[1])); ll prn=0; for(int i=tot,d;i;i--) { for(int t=0;t<=1;t++)ct[t][fa[d=q[i]]]+=ct[t][d]; prn+=(ll)(l[d]-l[fa[d]])*ct[0][d]*ct[1][d]; } printf("%lld\n",prn); return 0; }

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