bzoj2938【Poi2000】病毒

2938: [Poi2000]病毒

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 345  Solved: 176
[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

HINT

Source

这道题的思路很好

首先我们跑一次AC自动机，Trie树和失配边就构成了一个有向图。那么，能找到一个无限长的安全代码，当且仅当在非单词节点中存在环，用拓扑排序判断即可。

#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm>#include<cmath>#include<queue>#define F(i,j,n) for(int i=j;i<=n;i++)#define D(i,j,n) for(int i=j;i>=n;i--)#define ll long long#define pa pair<int,int>#define maxn 30100#define inf 1000000000using namespace std;struct edge_type{int next,to;}e[maxn*2];int go[maxn],in[maxn],head[maxn],t[maxn][2];int n,tot=1,cnt=0;bool v[maxn];char s[maxn];inline void add_edge(int x,int y){e[++cnt]=(edge_type){head[x],y};head[x]=cnt;}inline void insert(){scanf("%s",s);int len=strlen(s),now=1;F(i,0,len-1){int x=s[i]-'0';if (!t[now][x]) t[now][x]=++tot;now=t[now][x];}v[now]=true;}inline void bfs(){queue<int> q;q.push(1);while (!q.empty()){int x=q.front(),y,j;q.pop();v[x]|=v[go[x]];F(i,0,1){j=go[x];while (j&&!t[j][i]) j=go[j];if (t[x][i]){go[y=t[x][i]]=j?t[j][i]:1;q.push(y);}else t[x][i]=j?t[j][i]:1;}}}inline bool topsort(){queue<int> q;int sum=0;F(i,1,tot){if (v[i]) sum++;else F(j,0,1) if (!v[t[i][j]]){add_edge(i,t[i][j]);in[t[i][j]]++;}}F(i,1,tot) if (!v[i]&&!in[i]) q.push(i);while (!q.empty()){int x=q.front();q.pop();sum++;for(int i=head[x];i;i=e[i].next){int y=e[i].to;in[y]--;if (!in[y]) q.push(y);}}return sum==tot;}int main(){scanf("%d",&n);F(i,1,n) insert();bfs();printf("%s\n",topsort()?"NIE":"TAK");return 0;}