P2444 [POI2000]病毒(AC自动机)

Posted 2022-01-02 Harris-H

P2444 [POI2000]病毒(AC自动机)

多模式串匹配问题。

考虑建Trie图。

然后对于一个安全代码串，我们在Trie图进行匹配，显然因为无限长所以我们把Trie的边和fail 边 看成 单向边的话，必然是存在一个环，且该环能从根结点走到。

因此我们对每个模式串在Trie上的末尾结点标记，然后从根开始dfs找环即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=3e4+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
const int hashmod[4] = 402653189,805306457,1610612741,998244353;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define x first
#define y second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define per(i,a,b) for(int i=a;i>=b;--i)
#define ios ios::sync_with_stdio(false),cin.tie(nullptr) 
void Print(int *a,int n)
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\\n",a[n]); 

template <typename T>		//x=max(x,y)  x=min(x,y)
void cmx(T &x,T y)
	if(x<y) x=y;

template <typename T>
void cmn(T &x,T y)
	if(x>y) x=y;

int n;
char s[N];
struct ACAM
#define il inline
	int son[N][2],fa[N],id;
	bool ed[N];
	il void init()
		id=0,mst(fa,0),mst(son,0);
	
	il void ins(char *s)
		int rt = 0, n = strlen(s);
		for(int i=0;i<n;++i)
			int j=s[i]-'0';
			if(!son[rt][j]) son[rt][j]=++id;
			rt = son[rt][j];
		
		ed[rt] = true;
	
	il void bud()
		queue<int>q;
		for(int i=0;i<2;i++)
			if(son[0][i]) fa[son[0][i]]=0,q.push(son[0][i]);
		while(!q.empty())
			int u=q.front();q.pop();
			//printf("u=%d\\n",u);
			for(int i=0;i<2;i++)
				if(son[u][i])
			fa[son[u][i]] = son[fa[u]][i];
			q.push(son[u][i]);
			
				else son[u][i] = son[fa[u]][i];
			
			ed[u] |= ed[fa[u]];
			
	
	bool in[N],vis[N];
	void dfs(int u)
		in[u] = true;
		for(int i=0;i<2;i++)
			if(in[son[u][i]])
				puts("TAK");
				exit(0);
			
			else if(!ed[son[u][i]] && !vis[son[u][i]])
				vis[son[u][i]] =  true;
				dfs(son[u][i]);
			
		
		in[u] = false;
	
	il void solve()
		bud();
		dfs(0);
		puts("NIE");
		
T;
int main()
	scanf("%d",&n);
	rep(i,1,n)
		scanf("%s",s),T.ins(s);
	
	T.solve();
	return 0;