[BZOJ 2938][Poi2000]病毒

2938: [Poi2000]病毒


Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 424  Solved: 220
[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

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100010
using namespace std;
int debug;
int n;
char s[maxn];
int t[maxn][2];
bool val[maxn],vis[maxn];
int root,size,last[maxn],fail[maxn];
queue<int>Q;
int newnode(){
    return ++size;
}
void init(){
    root=newnode();
}
void insert(){
    int n=strlen(s+1),now=root;
    for(int i=1;i<=n;i++){
        int p=s[i]-'0';
        if(t[now][p])now=t[now][p];
        else t[now][p]=newnode(),now=size;
    }val[now]=1;
}
void buildfail(){
    for(int i=0;i<2;i++){
        if(!t[root][i])t[root][i]=root;
        else fail[t[root][i]]=root,Q.push(t[root][i]);
    }
    while(!Q.empty()){
        int now=Q.front();Q.pop();
        for(int i=0;i<2;i++){
            if(t[now][i]==0)t[now][i]=t[fail[now]][i];
            else fail[t[now][i]]=t[fail[now]][i],Q.push(t[now][i]);
        }last[now]=val[now]?now:last[fail[now]];
    }
}
int dp[maxn];
bool dfs(int now){
    if(last[now]||val[now]||dp[now])return false;
    if(vis[now])return true;
    vis[now]=true;
    if(dfs(t[now][0]))return true;
    if(dfs(t[now][1]))return true;
    vis[now]=false;
    dp[now]=true;
    return false;
}
 
int main(){
#ifndef ONLINE_JUDGE
    freopen("wir.in","r",stdin);
    freopen("wir.out","w",stdout);
#endif
    scanf("%d",&n);
    init();
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        insert();
    }
    buildfail();
    if(dfs(root))printf("TAK");
    else printf("NIE");
    return 0;
}