bzoj2938: [Poi2000]病毒

题目链接

bzoj2938: [Poi2000]病毒

题解

简单的来说，就是构造一个只含0和1的字符串，不能与Trie完成匹配。
dfs判一下trie图上是否存在与根节点相连没有串结尾的环
也就是有串结尾的点和fail指针指向串结尾的点，或指向不能走的点(也就是之前那一坨)的点也不能在环中

代码

#include<bits/stdc++.h> 
using namespace std; 
int n; 
const int maxn = 30007; 
char a[30007]; 
int ch[maxn][2],val[maxn]; 
int sz = 0; 
void Insert(char *s) { 
    int len = strlen(s + 1),now = 0 ; 
    for(int i = 1;i <= len;++ i) { 
        int c = s[i] - '0'; 
        if(!ch[now][c]) 
            ch[now][c] = ++ sz; 
        now = ch[now][c]; 
    } 
    val[now] = 1; 
} 
queue<int>q; 
int fail[maxn]; 
void getfail() { 
    for(int i = 0;i <= 1;++ i) if(ch[0][i]) q.push(ch[0][i]);
    while(!q.empty() ) { 
        int now = q.front(); 
        q.pop(); 
        for(int i = 0;i <= 1;++ i) { 
            if(ch[now][i]) {
                fail[ch[now][i]] = ch[fail[now]][i];
                val[ch[now][i]] |= val[ch[fail[now]][i]]; 
                q.push(ch[now][i]); 
            } 
            else ch[now][i]  = ch[fail[now]][i];    
        } 
    } 
} 
bool vis[maxn],ht[maxn]; 
void dfs(int x) { 
    vis[x] = 1; ht[x] = 1; 
    for(int i =0;i <= 1;++ i) { 
        int v = ch[x][i]; 
        if(v) { 
            if(vis[v]) { 
                puts("TAK"); 
                exit(0); 
            } 
            if(!ht[v] && !val[v]) dfs(v); 
        } 
    } 
    vis[x] = 0; 
} 
int main() { 
    scanf("%d",&n); 
    for(int i = 1;i <= n;++ i) { 
        scanf("%s",a + 1); 
        Insert(a); 
    } 
    getfail(); 
    dfs(0); 
    puts("NIE"); 
    return 0; 
}