bzoj 2938: [Poi2000]病毒 AC自動機

2019-11-08 01:41:12

字體：大中小

來源：轉載

供稿：網友

題意

二進制病毒審查委員會最近發現了如下的規律：某些確定的二進制串是病毒的代碼。如果某段代碼中不存在任何一段病毒代碼，那么我們就稱這段代碼是安全的。現在委員會已經找出了所有的病毒代碼段，試問，是否存在一個無限長的安全的二進制代碼。示例：例如如果{011, 11, 00000}為病毒代碼段，那么一個可能的無限長安全代碼就是010101…。如果{01, 11, 000000}為病毒代碼段，那么就不存在一個無限長的安全代碼。任務：請寫一個程序： l 讀入病毒代碼； l 判斷是否存在一個無限長的安全代碼； l 將結果輸出所有病毒代碼段的總長度不超過30000

分析

將所有字符建立AC自動機，若指針能在自動機上找到一個環則TAK。

代碼

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;char ch[30005];int fail[30005],trie[30005][3],n,val[30005],vis[30005],cnt,q[30005],inst[30005];void ins(){ int len=strlen(ch),now=1; for (int i=0;i<len;i++) if (trie[now][ch[i]-'0']) now=trie[now][ch[i]-'0']; else now=trie[now][ch[i]-'0']=++cnt; val[now]=1;}void get_fail(){ fail[0]=1; int head=1,tail=1;q[1]=1; while (head<=tail) { int u=q[head]; head++; for (int i=0;i<2;i++) { int v=trie[u][i]; if (!v) { if (u!=1) trie[u][i]=trie[fail[u]][i]; continue; } int k=fail[u]; while (!trie[k][i]) k=fail[k]; fail[v]=trie[k][i];val[v]|=val[trie[k][i]]; q[++tail]=v; } }}bool dfs(int x){ vis[x]=inst[x]=1; for (int i=0;i<2;i++) { if (inst[trie[x][i]]) return 1; if (!trie[x][i]||val[trie[x][i]]||vis[trie[x][i]]) continue; if (dfs(trie[x][i])) return 1; } inst[x]=0; return 0;}int main(){ trie[0][0]=trie[0][1]=cnt=1; scanf("%d",&n); for (int i=1;i<=n;i++) { scanf("%s",ch); ins(); } get_fail(); if (dfs(1))