BZOJ2938: [Poi2000]病毒(AC自动机)

#include<cstdio>
#include<queue>
#include<cstring>
#include<cstdlib>
using namespace std;
const int MAXN = 1e6 + 10, B = 2;
int N;
int ch[MAXN][2], fa[MAXN], fail[MAXN], flag[MAXN], tot = 0, root = 0;
char s[MAXN];
void insert(char *s) {
    int N = strlen(s + 1), now = root;
    for(int i = 1; i <= N; i++) {
        int nxt = s[i] - '0';
        if(!ch[now][nxt]) ch[now][nxt] = ++tot;
        now = ch[now][nxt];
    }
    flag[now] = 1;
}
void GetFail() {
    queue<int> q;
    for(int i = 0; i < B ;i++) 
        if(ch[root][i])
            fail[ch[root][i]] = root, q.push(ch[root][i]);
    while(!q.empty()) {
        int p = q.front(); q.pop();
        for(int i = 0; i < B; i++) {
            if(ch[p][i]) fail[ch[p][i]] = ch[fail[p]][i], flag[ch[p][i]] |= flag[fail[ch[p][i]]], q.push(ch[p][i]);
            else ch[p][i] = ch[fail[p]][i];
        }
    }
}
int f[MAXN], vis[MAXN];
void dfs(int x) {
    f[x] = 1;
    for(int i = 0; i < B; i++) {
        if(f[ch[x][i]] == 1) {puts("TAK"); exit(0);}
        else if(!flag[ch[x][i]] && f[ch[x][i]] != 2) dfs(ch[x][i]);
    }
    f[x] = 2;
}
int main() {
    scanf("%d", &N);
    for(int i = 1; i <= N; i++) {
        scanf("%s", s + 1);
        insert(s);
    }
    GetFail();
    dfs(root);
    puts("NIE");
    return 0;
} 
/*
3
011
11 
00000

*/