洛谷 P2444 [ POI 2000 ] 病毒 —— AC自动机+dfs

题目：https://www.luogu.org/problemnew/show/P2444

AC自动机上 dfs，不走结尾点，如果走出环就是有无限长度的串；

RE无数，原来是数组开成 2000 的了...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
int const xn=,xm=;
int n,cnt,c[xm][],fail[xm];
bool vis[xm],end[xm],in[xm];
char dc[xm];
queue<int>q;
void build()
{
  int nw=,l=strlen(dc);
  for(int i=;i<l;i++)
    {
      int t=dc[i]-'';
      if(!c[nw][t])c[nw][t]=++cnt; nw=c[nw][t];
    }
  end[nw]=;
}
void getfail()
{
  for(int i=;i<=;i++)
    if(c[][i])fail[c[][i]]=,q.push(c[][i]);
  while(q.size())
    {
      int x=q.front(); q.pop();
      end[x]|=end[fail[x]];
      for(int i=;i<=;i++)
        {
          if(c[x][i])fail[c[x][i]]=c[fail[x]][i],q.push(c[x][i]);
          else c[x][i]=c[fail[x]][i];
        }
    }
}
bool dfs(int x)
{
  vis[x]=; in[x]=;
  for(int i=;i<=;i++)
    {
      if(!c[x][i]||end[c[x][i]])continue;//也可没有 !c[x][i]
      if(in[c[x][i]])return ;//注意在判断 vis 之前
      if(vis[c[x][i]])continue;//!return 0!
      vis[c[x][i]]=;
      if(dfs(c[x][i]))return ;//
    }
  in[x]=;
  return ;
}
int main()
{
  scanf("%d",&n);
  for(int i=;i<=n;i++)scanf("%s",dc),build();
  getfail();
  if(dfs())printf("TAK\n");
  else printf("NIE\n");
  return ;
}