[后缀自动机 parent树] BZOJ 4566 [Haoi2016]找相同字符

对一个串A建自动机 另一个串B在上面匹配 考虑统计答案
对于当前匹配到的点，那么它parent树中的祖先代表的串现在肯定也出现了
对于每个出现的点，它代表了maxs[x]-maxs[fa[x]]个串
这些串出现了|right(x)|次，贡献就是|right(x)|*(maxs[x]-maxs[fa[x]])
因为要统计祖先的，所以把祖先的也累加到这个节点即可

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}

inline int read(char *s){
  char c=nc(); int len=0;
  for (;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
  for (;c>='a' && c<='z';s[++len]=c,c=nc()); s[++len]=0; return len-1;
}

const int N=250005;

struct state{
  int len,link,next[26];
}st[N<<1];
int ncnt,last;
int cnt[N<<1];

inline void Extend(char c){
  int cur=++ncnt,p; c-='a';
  st[cur].len=st[last].len+1; cnt[cur]=1;
  for (p=last;p!=-1 && !st[p].next[c];p=st[p].link)
    st[p].next[c]=cur;
  if (p==-1)
    st[cur].link=0;
  else{
    int q=st[p].next[c];
    if (st[q].len==st[p].len+1)
      st[cur].link=q;
    else{
      int nq=++ncnt;
      st[nq].len=st[p].len+1;
      st[nq].link=st[q].link;
      for (int i=0;i<26;i++) st[nq].next[i]=st[q].next[i];
      for (;p!=-1 && st[p].next[c]==q;p=st[p].link)
    st[p].next[c]=nq;
      st[q].link=st[cur].link=nq;
    }
  }
  last=cur;
}

int tmp[N<<1],sum[N<<1];

inline void Sort(){
  for (int i=0;i<=ncnt;i++) sum[st[i].len]++;
  for (int i=1;i<=st[last].len;i++) sum[i]+=sum[i-1];
  for (int i=0;i<=ncnt;i++) tmp[sum[st[i].len]--]=i;
}

char A[N],B[N];
int na,nb;

ll f[N<<1],Ans;

int main(){
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  na=read(A);
  st[0].link=-1;
  for (int i=1;i<=na;i++) Extend(A[i]);
  Sort();
  for (int i=ncnt+1;i>1;i--) cnt[st[tmp[i]].link]+=cnt[tmp[i]];
  for (int i=2,x;i<=ncnt+1;i++)
    x=tmp[i],f[x]=f[st[x].link]+(ll)cnt[x]*(st[x].len-st[st[x].link].len);
  nb=read(B);
  int p=0,l=0;
  for (int i=1;i<=nb;i++){
    while (p && !st[p].next[B[i]-'a'])
      p=st[p].link,l=st[p].len;
    if (st[p].next[B[i]-'a'])
      p=st[p].next[B[i]-'a'],l++;
    if (p){
      Ans+=f[st[p].link]+(ll)cnt[p]*(l-st[st[p].link].len);
    }
  }
  printf("%lld\n",Ans);
  return 0;
}