bzoj2938 [Poi2000]病毒

Description

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

• 读入病毒代码；
• 判断是否存在一个无限长的安全代码；
• 将结果输出

Input

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

Output

你应在在文本文件 $WIN.OUT$ 的第一行输出一个单词：

• $TAK$ ——假如存在这样的代码；
• $NIE$ ——如果不存在。

Sample Input

3
01
11
00000

Sample Output

NIE

Solution

一个无限长的合法串，就会在ac自动机上跑成一个环，且不会进入 $danger$ 节点。详见代码。

#include<bits/stdc++.h>
using namespace std;

#define N 1000001
#define rep(i, a, b) for (int i = a; i <= b; i++)

struct acDFAType {
    int ch[N][2], fail[N], tot;
    bool danger[N], inStk[N], vis[N];
    char s[N];
    void insert() {
        scanf("%s", s); int u = 0, len = strlen(s);
        rep(i, 0, len - 1) {
            if(!ch[u][s[i] - '0']) ch[u][s[i] - '0'] = ++tot;
            u = ch[u][s[i] - '0'];
        }
        danger[u] = 1;
    }
    void getFail() {
        register queue<int> q; rep(i, 0, 1) if(ch[0][i]) q.push(ch[0][i]);
        while(!q.empty()) {
            int u = q.front(); q.pop();
            rep(i, 0, 1) {
                int v = ch[u][i]; if(!v) { ch[u][i] = ch[fail[u]][i]; continue; } q.push(v);
                int tmp = fail[u]; while(tmp && !ch[tmp][i]) tmp = fail[tmp];
                fail[v] = ch[tmp][i]; danger[v] |= danger[fail[v]];
            }
        }
    }
    bool dfs(int u) {
        inStk[u] = 1;
        rep(i, 0, 1) {
            int v = ch[u][i]; if(inStk[v]) return 1; if(danger[v] || vis[v]) continue; vis[v] = 1;
            if(dfs(v)) return 1;
        }
        return inStk[u] = 0;
    }
}acm;

int main() {
    int n; cin >> n; while(n--) acm.insert(); acm.getFail(); puts(acm.dfs(0) ? "TAK" : "NIE");
    return 0;
}