BZOJ 2938 | AOJ 844 AC自动机 + dfs

题解

无限匹配， 说明从根节点出发必存在不经过模式串串尾的环

模式串串尾说明当前节点val值不为0

所以匹配不能经过串尾也不能经过fail指向节点为另一模式串串尾的节点

考虑吧fail合并， 节点保存自身和fail指向节点的val合并值, 搜索判环的时候只要

节点的val不为0， 就终止当前路径的搜索

没搜索到一个节点（val为0且未构成环）, 有两种搜索选择（顺配直接进入下一节点和

适配进入fail指针指向节点的下一节点， 所以可以在getFail时直接更新适配节点的当前匹配）
1. 判环
2. 合并fail
3. dfs + 记忆化标记

---

code

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
const int N = 30010;
int flag = 0;
struct  AC_Auto{
    int ch[N][2], f[N], val[N], sz, vis[N], mem[N];
    AC_Auto(){
        sz = 1;
        memset(ch, 0, sizeof ch);
        memset(f, 0, sizeof f);
        memset(val, 0, sizeof val);
        memset(vis, 0, sizeof vis);
        memset(mem, 0, sizeof mem);
    }
    void insert(string s){
        int u = 0;
        for(int i = 0; s[i]; ++i){
            int c = s[i] - '0';
            if(!ch[u][c]) ch[u][c] = sz++;
            u = ch[u][c];
        }
        val[u] = 1;
    }
    void getFail(){
        queue<int> q;
        q.push(0);
        while(!q.empty()){
            int t = q.front();
            q.pop();
            for(int c = 0; c < 2; ++c){
                if(ch[t][c]){
                    int u = ch[t][c], v = f[t];
                    q.push(u);
                    while(v && !ch[v][c]) v = ch[v][c];
                    f[u] = t ? ch[v][c] : 0;
                    val[u] |= val[f[u]];/**合并fail*/
                }else ch[t][c] = ch[f[t]][c];/**根据fail更新失配的匹配*/
            }
        }
    }
    void dfs(int x){
        if(vis[x]) {flag = 1; return;}/**找到环*/
        if(mem[x]||val[x])return;/**已被搜索未找到或者fai存在模式串被匹配*/
        vis[x]=1;
        mem[x] = 1;
        for(int i=0;i<2;i++)
        {
            dfs(ch[x][i]);

            if(flag)return;
        }
        vis[x]=0;/**清楚匹配*/
    }
}ac;

int n;
string s;
int main(){

    cin >> n;
    while(n--){
        cin >> s;
        ac.insert(s);
    }
    ac.getFail();
    ac.dfs(0);
    if(flag) cout << "TAK" << endl;
    else cout << "NIE" << endl;
    return 0;
}