bzoj2938(ac自动机)

2938: [Poi2000]病毒

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 421   Solved: 217
[ Submit][ Status][ Discuss]

Description

二进制病毒审查委员会最近发现了如下的规律：某些确定的二进制串是病毒的代码。如果某段代码中不存在任何一段病毒代码，那么我们就称这段代码是安全的。现在委员会已经找出了所有的病毒代码段，试问，是否存在一个无限长的安全的二进制代码。

示例：

例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。

任务：

请写一个程序：

l         读入病毒代码；

l         判断是否存在一个无限长的安全代码；

l         将结果输出

Input

 

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

Output

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

l         TAK——假如存在这样的代码；

l         NIE——如果不存在。

Sample Input

3
01
11
00000

Sample Output

NIE

解题思路：首先确定先构造trie树，然后可以这样想，存在无限长的数列表明永远匹配不到。

首先如果一个点的fail值是病毒节点，那么当前点也是不能的。

然后我可以这样构图，对于每个点，补全它的左右两点，于是就可以找经过这些补全的点是否存在不经过不能走的点的回路。

（一定要是补全的点）

因为原来的点，不可以判断。 但其实对于优化的ac自动机来说，不需要加点，因为每个点都有左右节点。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cstdlib>
using namespace std;
int n,size;
char c[31000];
int ch[64000][3];
int val[64000];
int q[64000];
int fail[64000],fa[64000],dui[64000];
bool mg[64000];
int vis[64000]; 
 
void build()
 {
    int lg=strlen(c); int now=0;
    for (int i=1;i<=lg;++i)
     {
        if (ch[now][int (c[i-1])-47]!=0)
         {
            now=ch[now][int(c[i-1])-47];
         }else
          { 
            ++size; ch[now][int(c[i-1])-47]=size;
            now=size; 
          }
     }
    val[now]=1;
 }
 
void check()
 {
    int tail=0; int head=0; ++tail; q[tail]=0;
    while (head<tail)
     {
        ++head;
        if (ch[q[head]][1]==0 && ch[q[head]][2]==0)
         {
            continue;
         }
        if (ch[q[head]][1]==0  || ch[q[head]][2]==0 )
         {
            ++size;
            if (ch[q[head]][1]==0) 
             {
                ch[q[head]][1]=size;
                ++tail; q[tail]=ch[q[head]][2];
              }else
               {
                ch[q[head]][2]=size; ++tail;
                q[tail]=ch[q[head]][1];
               }
         }else
          {
            ++tail; q[tail]=ch[q[head]][2];
            ++tail; q[tail]=ch[q[head]][1];
          }
     }
 }
 
void getfail()
 {
    int tail=0; int head=0; ++tail; q[tail]=0; fail[0]=fa[0]=0; 
    while (head<tail)
     {
        ++head;
        if (q[head]==0) fail[q[head]]=0;else
         {
            int now=fail[fa[q[head]]];
            while (now!=0 && ch[now][dui[head]]==0)
             {
               now=fail[now];
             }
            if (ch[now][dui[head]]!=0 && ch[now][dui[head]]!=q[head])
             {
                fail[q[head]]=ch[now][dui[head]];
             }
         }
        val[q[head]]|=val[fail[q[head]]];
        for (int i=1;i<=2;++i)
         if (ch[q[head]][i]!=0)
         {
            ++tail; q[tail]=ch[q[head]][i];
            fa[q[tail]]=q[head]; dui[tail]=i;
         }
      }
 }
 
void dfs(int o)
 {
    mg[o]=false;
    for (int i=1;i<=2;++i)
     if (ch[o][i]!=0)
      {
        if (mg[ch[o][i]]==false) 
         {
            cout<<"TAK";
            exit(0);
           }
        if (val[ch[o][i]]==1 || vis[ch[o][i]]) continue;
        vis[ch[o][i]]=1;
        dfs(ch[o][i]);
      }
    if (ch[o][1]==0 && ch[o][2]==0)
     {
      if (fail[o]!=0)
      {
      if (mg[fail[o]]==false)
      {
      cout<<"TAK";
      exit(0);
  }
if (val[fail[o]]==1 | vis[fail[o]]); else{
vis[fail[o]]=1; 
dfs(fail[o]);}
 }
}
    mg[o]=true;
 }
 
int main()
{
    int n;
    int size=0;
    scanf("%d ",&n);
    for (int i=1;i<=n;++i)
     {
        cin>>c;
        build();    
     }
    check();
    getfail();
    memset(mg,true,sizeof(mg));
    dfs(0);
    cout<<"NIE";
}