p2444 病毒

传送门

分析

先用AC自动机将所有字符串存起来,之后我们从根节点跑一遍dfs

判断时我们只需要判断这个环有没有任意一个字符串的结尾就可以判断是否安全了

注意一个节点的信息需要累加上它fail指针所指位置的信息

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
int trie[40000][2],fail[40000],sum[40000],cnt;
int vis[40000],mer[40000];
char s[40000];
inline void ins(){
    int n=strlen(s),i,p=0;
    for(i=0;i<n;i++){
      if(!trie[p][s[i]-'0'])trie[p][s[i]-'0']=++cnt;
      p=trie[p][s[i]-'0'];
    }
    sum[p]++;
}
queue<int>q;
inline void findfail(){
    int i,j,k,p=0;
    fail[p]=p;
    for(i=0;i<2;i++)
      if(trie[p][i]){
          fail[trie[p][i]]=p;
          q.push(trie[p][i]);
      }
    while(!q.empty()){
      int x=q.front();q.pop();
      for(i=0;i<2;i++)
        if(trie[x][i]){
          sum[trie[x][i]]+=sum[trie[fail[x]][i]];
          fail[trie[x][i]]=trie[fail[x]][i];
          q.push(trie[x][i]);
        }else trie[x][i]=trie[fail[x]][i];
    }
}
inline int dfs(int x){
    vis[x]=1;
    for(int i=0;i<2;i++){
      if(vis[trie[x][i]])return 1;
      if(mer[trie[x][i]]||sum[trie[x][i]])continue;
      mer[trie[x][i]]=1;
      if(dfs(trie[x][i]))return 1;
    }
    vis[x]=0;
    return 0;
}
int main(){
    int n,m,i,j,k;
    scanf("%d",&n);
    for(i=1;i<=n;i++){
      scanf("%s",s);
      ins();
    }
    findfail();
    if(dfs(0))puts("TAK");
      else puts("NIE");
    return 0;
}
posted @ 2018-11-03 19:52  水题收割者  阅读(241)  评论(0编辑  收藏  举报