BZOJ [Poi2000]病毒 AC自动机_DFS_细节

BZOJ [Poi2000]病毒 AC自动机_DFS_细节

Code:

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 40000
#define N 2 
using namespace std;
char str[maxn]; 
queue<int>Q; 
int ch[maxn][3], tag[maxn], f[maxn], vis[maxn], done[maxn];  
int tot,root=0; 
void ins(char p[])
{
    int n=strlen(p+1),cur=0; 
    for(int i=1;i<=n;++i)
    {
        int c=p[i]-'0';  
        if(!ch[cur][c]) ch[cur][c]=++tot; 
        if(tag[cur]) tag[ch[cur][c]]=1; 
        cur=ch[cur][c]; 
    } 
    tag[cur]=1;       
} 
void build()
{
    for(int i=0;i<2;++i) if(ch[root][i]) Q.push(ch[root][i]); 
    while(!Q.empty())
    {
        int u=Q.front();Q.pop(); 
        for(int i=0;i<N;++i)
        {
            int q=ch[u][i]; 
            if(!q) 
            {
                ch[u][i]=ch[f[u]][i]; 
                continue; 
            }
            f[q]=ch[f[u]][i], tag[q]|=tag[f[q]];   
            Q.push(q); 
            if(tag[u])  
                tag[q]=1; 
        }         
    }
}  
void dfs(int u)
{
    if(vis[u]) 
    { 
        printf("TAK\n"); 
        exit(0); 
    }
    if(tag[u] || done[u]) return;
    done[u]=vis[u]=1; 
    dfs(ch[u][0]); 
    dfs(ch[u][1]); 
    vis[u]=0; 
}
int main()
{
   //  setIO("input"); 
    int tot; 
    scanf("%d",&tot);
    while(tot--) scanf("%s",str+1), ins(str);  
    build(); 
    dfs(0); 
    printf("NIE\n"); 
    return 0;  
}

  

posted @ 2019-06-10 15:01 EM-LGH 阅读( ...) 评论( ...) 编辑 收藏
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值