[POI2000]病毒(AC自动机,简单环搜索)

题意

判断是否能够构建一个字符串(无限长),使得不含个给定的字串.

数据范围:

题解

涉及多个串的匹配,可以考虑使用自动机.但通常的自动机是用于给定串的匹配.而该题要求我们构建符合条件的串.

这里就涉及一个对自动机构建指针的理解:将树构建成图.

由于要构建不含给定串,故对每个给定串终止位打上标记.

这里要注意:如果每个结点被打上标记,那么它会使指针指向它的结点被打上标记

如果我们能够在图上找到一个符合条件的环,那么就可以构建无限长的目标串.

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;

struct Trie {
  int next[2];
  int fail;
  bool virus;
} trie[5000005];
int cnt_trie;
void insert(char *s);
void getfail();
bool flag;
bool vis[500005];
void dfs(int);
int N;
char s[30005];

int main(){
  int i;
  scanf("%d", &N);
  for(i = 1; i <= N; i++){
    scanf("%s", s);
    insert(s);
  }
  getfail();
  dfs(0);
  if(flag) printf("TAK\n");
  else printf("NIE\n");
  return 0;
}

void dfs(int u){
  if(flag) return;
  if(vis[u]){
    flag = true;
    return;
  }
  int i;
  vis[u] = true;
  for(i = 0; i < 2; i++){
    int v = trie[u].next[i];
    if(trie[v].virus) continue;
    dfs(v);
  }
  vis[u] = false;
}

void insert(char *s){
  int id, root = 0;
  while(*s){
    id = *s - '0';
    if(!trie[root].next[id])
      trie[root].next[id] = ++cnt_trie;
    root = trie[root].next[id];
    s++;
  }
  trie[root].virus = true;
}

void getfail(){
  int i;
  queue<int> Q;
  for(i = 0; i < 2; i++)
    if(trie[0].next[i])
      Q.push(trie[0].next[i]);

  while(!Q.empty()){
    auto u = Q.front(); Q.pop();

    for(i = 0; i < 2; i++){
      int v = trie[u].next[i];
      if(v){
        Q.push(v);
        trie[v].fail = trie[trie[u].fail].next[i];
        trie[v].virus |= trie[trie[v].fail].virus;
      }
      else
        trie[u].next[i] = trie[trie[u].fail].next[i];
    }
  }
}