洛古2444 [POI2000]病毒 ac自動機

題目大意

二進制病毒審查委員會最近發現瞭如下的規律：某些確定的二進制串是病毒的代碼。如果某段代碼中不存在任何一段病毒代碼，那麼我們就稱這段代碼是安全的。現在委員會已經找出了所有的病毒代碼段，試問，是否存在一個無限長的安全的二進制代碼。

示例：

例如如果{011, 11, 00000}爲病毒代碼段，那麼一個可能的無限長安全代碼就是010101…。如果{01, 11, 000000}爲病毒代碼段，那麼就不存在一個無限長的安全代碼。

任務：

請寫一個程序：

1.在文本文件WIR.IN中讀入病毒代碼；

2.判斷是否存在一個無限長的安全代碼；

3.將結果輸出到文件WIR.OUT中。

分析

既然這個自動機又像一個圖，那我們的問題不就變成了——在AC自動機(trie圖)中尋找一個環，並且環上沒有任何危險節點，並且還要注意，這個環能被根節點訪問到（也就是說從根節點出發能在不經過危險節點的情況下走到到這個環

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 50000
struct Node
{
    int vis[2];
    int fail,lt;
}t[MAX];
int n,tot;
char ch[MAX];
void insert(char *s)
{
    int l=strlen(s+1),now=0;
    for(int i=1;i<=l;++i)
    {
        if(!t[now].vis[s[i]-48])
            t[now].vis[s[i]-48]=++tot;
        now=t[now].vis[s[i]-48];
    }
    t[now].lt=1;
}
void Getfail()
{
    queue<int> Q;
    for(int i=0;i<2;++i)
        if(t[0].vis[i])Q.push(t[0].vis[i]);
    while(!Q.empty())
    {
        int u=Q.front();Q.pop();
        t[u].lt|=t[t[u].fail].lt;
        for(int i=0;i<2;++i)
            if(t[u].vis[i])
                t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(t[u].vis[i]);
            else t[u].vis[i]=t[t[u].fail].vis[i]; 
    }
}
bool vis[MAX];
bool vv[MAX];
void DFS(int u)
{
    if(t[u].lt)return;
    if(vis[u]){puts("TAK");exit(0);}
    if(vv[u])return;
    vis[u]=true;vv[u]=true;
    for(int i=0;i<2;++i)
        if(t[u].vis[i])
            DFS(t[u].vis[i]);
    vis[u]=false;
}
int main()
{
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",ch+1);
        insert(ch);
    }
    Getfail();
    DFS(0);
    puts("NIE");
    return 0;
}