【BZOJ2938】病毒，AC自动机练习

传送门（权限题）

---

2938: [Poi2000]病毒

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 462 Solved: 240
[Submit][Status][Discuss]
Description

二进制病毒审查委员会最近发现了如下的规律：某些确定的二进制串是病毒的代码。如果某段代码中不存在任何一段病毒代码，那么我们就称这段代码是安全的。现在委员会已经找出了所有的病毒代码段，试问，是否存在一个无限长的安全的二进制代码。
示例：
例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。
任务：
请写一个程序：
l 读入病毒代码；
l 判断是否存在一个无限长的安全代码；
l 将结果输出
Input

第一行包括一个整数n，表示病毒代码段的数目。以下的n行每一行都包括一个非空的01字符串——就是一个病毒代码段。所有病毒代码段的总长度不超过30000。
Output

你应在在文本文件WIN.OUT的第一行输出一个单词：
l TAK——假如存在这样的代码；
l NIE——如果不存在。
Sample Input

3

01

11

00000

Sample Output

NIE

---

写在前面：给了三个样例的良心题
思路：
我们首先确定，如果可能，那么符合条件的字符串中一定有一个循环的字符串，如果不可能，那一定没有循环的字符串（因为循环的字符串是有循环节的，如果没有循环的字符串，那就意味着所有的循环节都不可以，那就是没有符合条件的了，反之亦如此）
然后偷个懒

zky:

首先我们把所有串建一个AC自动机

方便起见我们直接把fail指针合并到子结点

如果一个串能无限长,也就是说它可以在AC自动机上一直进行匹配但就是匹配不上

也就是说匹配指针不能走到val为1的结点,设这个点为x

即root..x是一个病毒串

那么fail指针指向x的y也不能走

因为root..x是root..y的一个后缀

处理出来判断有向图是否有环

dfs即可

建fail时把那些当前节点i没有的字符（相当于失配状态）都转移到fail[i]上去，之后我们只要从根上出发dfs一通乱走，如果途中遇到字符串结束的标记节点，就不走（因为一走不就相当于有这个病毒了吗），如果最后无路可走（怎么选都要经过标记节点）那就NiE了，但如果我们能走出一个没有标记节点的环，那就说明我们可以找到一个循环节使字符串合法且无限延长下去了
注意：
DFS时注意及时退出，减少搜索树深度，用两个bool数组记录节点的搜索状态，其中一个表示这个点是否曾经搜索过，减少重复搜索；另一个表示这个点是否在我们目前想要的环的路径上（可能有点抽象，就是说我们是否已经经过这个点并且走在这个点所连接的边上，搜索完是要回溯的，而前一个bool数组只是单纯记录这个点是否走过，不必回溯更改）
代码：

#include<bits/stdc++.h>using namespace std;int root=1,tot=1,n;int trie[30002][2],fail[30002];bool num[30002],vis[30002],flag[30002];char s[30002];queue<int>q;void insert(char s[]){    int len=strlen(s),now=root;    for (int i=0;i<len;i++)    {        if (!trie[now][s[i]-'0']) trie[now][s[i]-'0']=++tot;        now=trie[now][s[i]-'0'];    }    num[now]=1;}void build(){    int now,tmp;    q.push(root);    while (!q.empty())    {        now=q.front();        q.pop();        for (int i=0;i<2;i++)        if (trie[now][i])        {            tmp=fail[now];            while (tmp&&!trie[tmp][i]) tmp=fail[tmp];            if (tmp&&now!=root)                fail[trie[now][i]]=trie[tmp][i],                num[trie[now][i]]+=num[trie[tmp][i]];            else fail[trie[now][i]]=root;            q.push(trie[now][i]);        }        else        {            if (now==root) trie[now][i]=root;            else trie[now][i]=trie[fail[now]][i];        }    }}void dfs(int x){    if (flag[x]) {printf("TAK");exit(0);}    if (num[x]||vis[x]) return;    if (x!=root)vis[x]=1;    flag[x]=1;    for (int i=0;i<2;i++)        dfs(trie[x][i]);    flag[x]=0;}main(){    scanf("%d",&n);    for (int i=1;i<=n;i++)        scanf("%s",s),        insert(s);    build();    dfs(root);    printf("NIE");}