bzoj 4566 [Haoi2016]找相同字符——广义后缀自动机

题目：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+,M=8e5+,K=;
int tot=,go[M][K],fa[M],l[M],ct[][M],ans[M];
char s[][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]+;
  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]+)return q; return cz(p,w);
    }
  int np=++tot;l[np]=l[p]+;
  for(;p&&!go[p][w];p=fa[p])go[p][w]=np;
  if(!p)fa[np]=;
  else
    {
      int q=go[p][w];
      if(l[q]==l[p]+)fa[np]=q;
      else fa[np]=cz(p,w);
    }
  return np;
}
void Rsort(int n)
{
  for(int i=;i<=tot;i++)tx[l[i]]++;
  for(int i=;i<=n;i++)tx[i]+=tx[i-];
  for(int i=;i<=tot;i++)q[tx[l[i]]--]=i;
}
int main()
{
  scanf("%s",s[]);scanf("%s",s[]);
  int n[];n[]=strlen(s[]);n[]=strlen(s[]);
  for(int t=;t<=;t++)
    for(int pr=,i=;i<n[t];i++)pr=ins(pr,s[t][i]-'a'),ct[t][pr]=;
  Rsort(max(n[],n[])); ll prn=;
  for(int i=tot,d;i;i--)
    {
      for(int t=;t<=;t++)ct[t][fa[d=q[i]]]+=ct[t][d];
      prn+=(ll)(l[d]-l[fa[d]])*ct[][d]*ct[][d];
    }
  printf("%lld\n",prn);
  return ;
}