bzoj 2938 [Poi2000]病毒

2938: [Poi2000]病毒
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 849 Solved: 438
[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

---

【分析】
我真傻，真的
AC自动机+dfs
http://www.cnblogs.com/DaD3zZ-Beyonder/p/6139083.html

---

【代码】

//bzoj 2938 [Poi2000]病毒 
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
using namespace std;
const int mxn=1000005;
bool vis[mxn],bo[mxn];
queue <int> q;
int n,num;
char s[mxn];
struct node
{
    int son[2],fail,cnt;
}a[mxn];
inline void clear(int x)
{
    a[x].son[0]=a[x].son[1]=a[x].fail=a[x].cnt=0;
}
inline void trie()
{
    int x=0,len=strlen(s);
    fo(i,0,len-1)
    {
        int t=s[i]-'0';
        if(!a[x].son[t])
        {
            num++;
            clear(num);
            a[x].son[t]=num;
        }
        x=a[x].son[t];
    }
    a[x].cnt++;
}
inline void build()
{
    while(!q.empty()) q.pop();
    fo(i,0,1)
      if(a[0].son[i]) q.push(a[0].son[i]);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        int fail=a[x].fail;
        fo(i,0,1)
        {
            int y=a[x].son[i];
            if(y)
            {
                a[y].fail=a[fail].son[i];
                a[y].cnt|=a[a[y].fail].cnt;
                q.push(y);
            }
            else a[x].son[i]=a[fail].son[i]; //???
        }
    }
}
inline bool dfs(int x)
{
    vis[x]=1;
    fo(i,0,1)
    {
        int v=a[x].son[i];
        if(vis[v]) return 1;
        if(bo[v] || a[v].cnt) continue;
        bo[v]=1;
        if(dfs(v)) return 1;
    }
    vis[x]=0;
    return 0;
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        scanf("%s",s);
        trie();
    }
    build();
    if(dfs(0)) printf("TAK\n");
    else printf("NIE\n");
    return 0;
}
/*
3
00
01
00000
*/