【[POI2000]病毒】
\(Ac\)自动机好题了
这个题要求我们一直无法匹配到结束标记,所以我们直接在\(trie\)图上找到一个环,这个环可以被根节点到达,之后还没有结束标记
发现自己不会\(dfs\)找环,于是直接莽上\(tarjan\)
之后一直写挂各种\(sb\)
代码
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#define re register
#define LL long long
#define maxn 30005
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
struct E{int v,nxt;}e[maxn<<4];
int head[maxn];
int n,m,cnt,top,tot,p,num,mid;
char S[maxn];
int son[maxn][2],fail[maxn],flag[maxn];
int dfn[maxn],low[maxn],f[maxn],st[maxn],col[maxn],d[maxn],vis[maxn];
inline void ins()
{
scanf("%s",S+1);
int len=strlen(S+1),now=0;
for(re int i=1;i<=len;i++)
{if(!son[now][S[i]-'0']) son[now][S[i]-'0']=++cnt;now=son[now][S[i]-'0'];}
flag[now]=1;
}
inline void Build()
{
std::queue<int> q;
for(re int i=0;i<2;i++) if(son[0][i]) q.push(son[0][i]);
while(!q.empty())
{
int k=q.front();q.pop();
flag[k]|=flag[fail[k]];
for(re int i=0;i<2;i++)
if(son[k][i]) fail[son[k][i]]=son[fail[k]][i],q.push(son[k][i]);
else son[k][i]=son[fail[k]][i];
}
}
void tarjan(int x)
{
if(flag[x]) return;
f[x]=1,st[++top]=x,dfn[x]=low[x]=++tot;
for(re int i=0;i<2;i++)
{
if(flag[son[x][i]]) continue;
if(!dfn[son[x][i]]) tarjan(son[x][i]),low[x]=min(low[x],low[son[x][i]]);
else if(f[son[x][i]]) low[x]=min(low[x],dfn[son[x][i]]);
}
if(dfn[x]==low[x])
{
++p;int t=0;
do {mid=st[top--];f[mid]=0,t++,col[mid]=p;}while(x!=mid);
d[p]=t;
}
}
inline void add_edge(int x,int y){e[++num].v=y,e[num].nxt=head[x],head[x]=num;}
int dfs(int x)
{
if(d[x]>1) return 1;
int ff=0;
vis[x]=1;
for(re int i=head[x];i;i=e[i].nxt)
if(!vis[e[i].v]) ff|=dfs(e[i].v);
return ff;
}
int main()
{
scanf("%d",&n);
for(re int i=1;i<=n;i++) ins();Build();
for(re int i=0;i<=cnt;i++) if(!dfn[i]) tarjan(i);
for(re int i=0;i<=cnt;i++)
for(re int j=0;j<2;j++) if(col[son[i][j]]!=col[i]) add_edge(col[i],col[son[i][j]]);
for(re int i=0;i<=cnt;i++)
for(re int j=0;j<2;j++)
if(son[i][j]==i) d[col[i]]++;
vis[0]=1;
if(dfs(col[0])) puts("TAK");else puts("NIE");
return 0;
}