bzoj2938 poi病毒 AC自动机

时间:2022-03-21 22:12:16

题目传送门

思路:

  要求构建一个字符串,使得这个字符串不包含给出的任意一个单词。

  如果我们已经构建出了一个安全代码,放在ac自动机上跑,那么我们必定不能得到任何一个字符串,此时我们得到的fail指针必定是在一个环上循环,并且这个环不包含单词的末尾。

  我们也知道fail指针最后是会指回0点的,那么此时我们其实就是要求一个环,这个换不包含单词末尾即可。加一个优化就是,如果fail指针指向的地方是末尾,那么这个地方等同于末尾。

#include<bits/stdc++.h>
#define clr(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=;
char s[maxn],p[maxn];
int trie[maxn][],cntword[maxn],fail[maxn],cnt=;
int n;
bool vis[maxn],ins[maxn];
void insert(char *s){
int root=;
int si=strlen(s);
for(int i=;i<si;i++)
{
int Next=s[i]-'';
if(!trie[root][Next])trie[root][Next]=++cnt;
root=trie[root][Next];
}
cntword[root]++;
}
void getfail(){
queue<int >q;
for(int i=;i<=;i++)
{
if(trie[][i]){
q.push(trie[][i]);
}
}
while(!q.empty())
{
int now=q.front();
q.pop();
for(int i=;i<=;i++)
{
if(trie[now][i]){
fail[trie[now][i]]=trie[fail[now]][i];
if(cntword[fail[trie[now][i]]])cntword[trie[now][i]]=;//如果fail指针指向的是病毒的末尾,那么这个也是病毒
q.push(trie[now][i]);
}else{
trie[now][i]=trie[fail[now]][i];
}
}
}
} void init(){
clr(trie,);
clr(cntword,);
clr(ins,),clr(vis,);
}
bool dfs(int u)
{
ins[u]=;
for(int i=;i<=;i++)
{
int v=trie[u][i];
if(ins[v]==)return true;
if(vis[v]||cntword[v])continue;
vis[v]=;
if(dfs(v))return true;
}
ins[u]=;
return false;
}
int main(){
scanf("%d",&n);
for(int i=;i<=n;i++)
{
scanf("%s",p);
insert(p);
}
fail[]=;
getfail();
if(dfs())puts("TAK");
else puts("NIE"); }