洛谷2444:病毒

时间:2020-01-25 22:32:26   收藏:0   阅读:90

洛谷2444:病毒

题意:

思路:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 3e4 + 10;
char s[maxn];
int n;

struct AC_Automaton
{
    int trie[maxn][5];
    int val[maxn];
    int fail[maxn];
    int tot;

    void ins(char *str)
    {
        int len = strlen(str), p = 0;
        for(int k = 0; k < len; k++)
        {
            int ch = str[k] - '0';
            if(trie[p][ch] == 0) trie[p][ch] = ++tot;
            p = trie[p][ch];
        }
        val[p] = 1;
    }

    void build()
    {
        queue<int> q;
        for(int i = 0; i < 2; i++)
        {
            if(trie[0][i])
            {
                //第二层指向根节点
                fail[trie[0][i]] = 0;
                q.push(trie[0][i]);
            }
        }
        while(q.size())
        {
            int x = q.front(); q.pop();
            for(int i = 0; i < 2; i++)
            {
                if(trie[x][i])
                {
                    fail[trie[x][i]] = trie[fail[x]][i];
                    val[trie[x][i]] |= val[fail[trie[x][i]]];
                    q.push(trie[x][i]);
                }
                else trie[x][i] = trie[fail[x]][i];
            }
        }
    }

    bool v1[maxn];
    
    void dfs(int x)
    {
        if(v1[x])
        {
            puts("TAK");
            exit(0);
        }
        if(val[x]) return;
        v1[x] = 1;
        if(trie[x][0]) dfs(trie[x][0]);
        if(trie[x][1]) dfs(trie[x][1]);
        v1[x] = 0;
    }
}AC;


int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
    {
        scanf("%s", s);
        AC.ins(s);
    } AC.build();
    AC.dfs(0); //从字典树根节点开始找环
    puts("NIE"); //不存在无限的串
    return 0;
}

原文:https://www.cnblogs.com/zxytxdy/p/12233485.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!