【Codeforces 1207G】Indie Album（AC自动机）

题目链接

听说这学期要学AC自动机所以来复习一下qwq

对$T$串构建AC自动机，求出fail树和转移图（$to[x][i]$表示$x$节点经过字符$i$到达的状态）。对于一个$S$串，从自动机的起点开始按转移图逐字符转移，那么$T$串在$S$串中出现的次数，就是$T$串终结点的fail子树内的点在$S$串的转移过程中被到达的次数（重复到达算多次）。注意到$S$串的构建是树形结构，我们把$Si$的询问挂在对应树节点，在这棵树上DFS。转移图上每个点有个计数器，初始时为$0$；DFS进入一个节点时，对应的转移图上节点计数器$+1$，退出时$-1$。容易发现，到达一点并做$+1$操作后，有且只有根到该点的一条链在对应转移图节点上做了$+1$，于是可以解决所有$Si$有关的询问，只要求对应$T$串终结点fail树子树内计数器的和即可。可以用DFS序+树状数组维护。总复杂度$O(26\ast len(T)+n\ast log(len(T)))$。

一开始我犯了一个错误，就是对于$son[x][i]$不存在的点，我在求$to[x][i]$时还是用从$fail[x]$一直往上跳的方法。如果$x$的深度是$D$，我这样的一次操作相当于凭空添加了一个长度$D+1$的串，所以复杂度不对。学习了一下正确姿势：如果$son[x][i]$存在，$to[x][i]$即为$son[x][i]$，且$son[x][i]$的$fail$为$to[fail[x]][i]$；如果$son[x][i]$不存在，$to[x][i]$即为$to[fail[x]][i]$。这样每个$to[x][i]$就都是$O(1)$求辣。为了节省空间可不必开$to$数组，直接在构建自动机的过程中将$son$数组改写为$to$数组即可。

const int N=400005;
int n,m;
//Trie & Fail Tree
int son[N][26],cnt;char S[N];
int insert( )
{
  int i,x=0,K=strlen(S+1);
  for(i=1;i<=K;i++)
    {
      int &y=son[x][S[i]-'a'];
      if(!y) y=++cnt;x=y;
    }
  return x;
}
int fail[N];
vector<int>tr[N];
int L[N],R[N],tim;
void DFS(int x)
{
  L[x]=++tim;
  for(auto &y:tr[x]) DFS(y);
  R[x]=tim;
}
void build( )
{
  queue<int>q;
  int i,x,y,k;
  for(i=0;i<26;i++)
    if(son[0][i]) q.push(son[0][i]);
  while(!q.empty( ))
    {
      x=q.front( );q.pop( );
      for(i=0;i<26;i++)
		{
		  y=son[x][i];
		  /*
		  k=fail[x];
		  while(k&&!son[k][i]) k=fail[k];
		  if(y) fail[y]=son[k][i],q.push(y);
		  else son[x][i]=son[k][i];
		  */
		  if(y) fail[y]=son[fail[x]][i],q.push(y);
		  else son[x][i]=son[fail[x]][i];
		}
    }
  for(i=1;i<=cnt;i++)
    tr[fail[i]].push_back(i);
  DFS(0);
}
//dfs & fenwick
int a[N],qt[N],ans[N];char X[N];
void add(int x) {for(int i=x;i<=tim;i+=i&-i) a[i]++;}
void del(int x) {for(int i=x;i<=tim;i+=i&-i) a[i]--;}
int ask(int x) {int i,r=0;for(i=x;i;i-=i&-i) r+=a[i];return r;}
int ask(int l,int r) {return ask(r)-ask(l-1);}
vector<int>E[N],Q[N];
void dfs(int x,int rx)
{
  add(L[rx]);
  for(auto &y:Q[x]) ans[y]=ask(L[qt[y]],R[qt[y]]);
  for(auto &y:E[x]) dfs(y,son[rx][X[y]-'a']);
  del(L[rx]);
}
int main( )
{
  int i,j,x;
  read(n);
  for(i=1;i<=n;i++)
    {
      x=read( );
      if(x==1) E[0].push_back(i);
      else j=read( ),E[j].push_back(i);
      X[i]=getc( );
    }
  read(m);
  for(i=1;i<=m;i++)
    {
      x=read( );
      scanf("%s",S+1);
      qt[i]=insert( );
      Q[x].push_back(i);
    }
  build( );
  dfs(0,0);
  for(i=1;i<=m;i++) printf("%d\n",ans[i]);
  return 0;
}