[POI2000]病毒

　　　　　　    [POI2000]病毒

题目描述

二进制病毒审查委员会最近发现了如下的规律：某些确定的二进制串是病毒的代码。如果某段代码中不存在任何一段病毒代码，那么我们就称这段代码是安全的。现在委员会已经找出了所有的病毒代码段，试问，是否存在一个无限长的安全的二进制代码。

示例：

例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。

任务：

请写一个程序：

1.在文本文件WIR.IN中读入病毒代码；

2.判断是否存在一个无限长的安全代码；

3.将结果输出到文件WIR.OUT中。

输入输出格式

输入格式：

在文本文件WIR.IN的第一行包括一个整数n(nle 2000)(n≤2000) ，表示病毒代码段的数目。以下的n行每一行都包括一个非空的01字符串——就是一个病毒代码段。所有病毒代码段的总长度不超过30000。

输出格式：

在文本文件WIR.OUT的第一行输出一个单词：

TAK——假如存在这样的代码；

NIE——如果不存在。

输入输出样例

输入样例#1：

3
01 
11 
00000

输出样例#1：

NIE

根据给出的病毒建出AC自动机，在上面DFS找环，单词的结尾的节点不能走，有环就代表有解。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define LL long long

using namespace std;

queue<int> q;
int len;
const int MAXN = 2e3 + 10;
char s[MAXN];
int vis[MAXN], instack[MAXN];
int flag = 0;
struct trie {
    int fail;
    int vis[2];
    int judge;
} t[MAXN * 10];

inline LL read()
{
    LL x = 0, w = 1; char ch = 0;
    while(ch < '0' || ch > '9') {
        if(ch == '-') {
            w = -1;
        }
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9') {
        x = x * 10 + ch - '0';
        ch = getchar();    
    }
    return x * w;
}
int tot = 0;

void insert()
{
    int x = 0;
    for(int i = 1; i <= len; i++) {
        if(t[x].vis[s[i] - '0'] == 0) {
            t[x].vis[s[i] - '0'] = ++tot;
        }
        x = t[x].vis[s[i] - '0'];
    }
    t[x].judge = 1;
}

void getfail()
{
    for(int i = 0; i < 2; i++) {
        if(t[0].vis[i]) {
            t[t[0].vis[i]].fail = 0;
            q.push(t[0].vis[i]);        
        }
    }
    while(!q.empty()) {
        int x = q.front();
        q.pop();
        for(int i = 0; i < 2; i++) {
            if(!t[x].vis[i]) {
                t[x].vis[i] = t[t[x].fail].vis[i];
            } else { 
                t[t[x].vis[i]].fail = t[t[x].fail].vis[i];
                if(t[t[t[x].vis[i]].fail].judge == 1) {
                    t[t[x].vis[i]].judge = 1;
                }
                q.push(t[x].vis[i]);
            }
        }
    }
}

void DFS(int x)
{
    instack[x]++;
    if(flag || t[x].judge == 1) {
        return;
    }
    vis[x] = 1;
    for(int i = 0; i < 2; i++) {
        if(vis[t[x].vis[i]] == 1 && instack[t[x].vis[i]]) {
            //cout<<t[x].vis[i]<<endl;
            flag = 1;
            return;
        }
        if(!vis[t[x].vis[i]]) {
            DFS(t[x].vis[i]);
        }
    }
    instack[x]--;
}

int main()
{
    int n;
    n = read();
    for(int i = 1; i <= n; i++) {
        scanf("%s", s + 1);
        len = strlen(s + 1);
        insert();
    }
    getfail();
    flag = 0;
    DFS(0);
    if(flag) {
        cout<<"TAK"<<endl;
    } else {
        cout<<"NIE"<<endl;
    }
}