bzoj2938: [Poi2000]病毒-CSDN博客

被Star_Feel大爷带着做题

明显大力AC机然后找环

本来我一开始想的是先去有另一个病毒为前缀的病毒，结果今天早上写的时候偷懒没写

结果跳fail的时候会跳到中间。。。无语，Star_Feel大爷叫我son或一下now和fail就过了

还有神仙是直接把fail接到儿子的更流弊了。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

struct Trie
{
    int w[2],fail; bool ed;
}tr[31000];int trlen;
char ss[31000];
void insert()
{
    int now=0; int len=strlen(ss+1);
    for(int i=1;i<=len;i++)
    {
        int x=ss[i]-'0';
        if(tr[now].w[x]==0)
            tr[now].w[x]=++trlen;
        now=tr[now].w[x];
    }
    tr[now].ed=true;
}
int list[31000];
void bfs()
{
    int head=1,tail=2;list[1]=0;
    while(head!=tail)
    {
        int now=list[head];
        for(int x=0;x<=1;x++)
        {
            int son=tr[now].w[x];
            if(son==0)continue;
            tr[son].ed|=tr[now].ed;
            if(now==0)tr[son].fail=0;
            else
            {
                int p=tr[now].fail;
                while(p!=0&&tr[p].w[x]==0)p=tr[p].fail;
                tr[son].fail=tr[p].w[x];
                tr[son].ed|=tr[tr[son].fail].ed;
            }
            
            list[tail++]=son;
        }
        head++;
    }
}

//------------------------AC---------------------------------------

int v[31000];
bool dfs(int now)
{
    if(tr[now].ed==true)return false;
    if(v[now]>=0)return v[now];
    v[now]=1;
    for(int x=0;x<=1;x++)
    {
        int p=now;
        while(p!=0&&tr[p].w[x]==0)p=tr[p].fail;
        if(dfs(tr[p].w[x]))return true;
    }
    v[now]=0;
    return false;
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%s",ss+1), insert();
    bfs();
    
    memset(v,-1,sizeof(v));
    if(dfs(0))printf("TAK\n");
    else printf("NIE\n");
    return 0;
}