2938: [Poi2000]病毒

考虑如果已经有了一个无限长的串,那么现在就是判断这个串能不能合法。
对于所有串建立ac自动机,那么显然如果一个点是病毒,其以后的点都是病毒,
并且如果一个点的fail是病毒,那么这个点也应该是病毒。
在ac自动机上跑,如果能一直不经过所有标记过的点(即为环)那么一定存在这个串。

c++代码如下:

#include
#define rep(i,x,y) for(register int i = x; i <= y;++ i)
#define repd(i,x,y) for(register int i = x; i >= y;-- i)
using namespace std;

const int N = 3e4+500;
int n,sz,fail[N],t[N][2];
char s[N]; bool val[N];

inline void insert(char*s)
{
    int n = strlen(s + 1);
    int x = 1;
    rep(i,1,n)
    {
        if(val[x]) return;
        if(!t[x][s[i] - '0']) t[x][s[i] - '0'] = ++ sz;
        x = t[x][s[i] - '0'];
    }

    val[x] = 1;
}

int q[N],st,en;
inline void build_ac()
{
    t[0][0] = t[0][1] = 1;
    q[0] = 1;
    while(st <= en)
    {
        int x = q[st++];
        rep(i,0,1)
            if(t[x][i]){
                int y = fail[x];
                while(!t[y][i]) y = fail[y];
                fail[t[x][i]] = t[y][i];
                val[t[x][i]] |= val[t[y][i]];
                q[++en] = t[x][i];
        }
    }
}

void pre(int x,bool e)
{
    e |= val[x];val[x] |= e;
    rep(i,0,1) if(t[x][i]) pre(t[x][i],e);
}

bool vis[N],inq[N];
bool dfs(int x)
{
    if(inq[x]) return true;
    if(vis[x]) return false;
    vis[x] = 1; inq[x] = 1;
    rep(i,0,1)
        if(t[x][i] && !val[t[x][i]]){
            if(dfs(t[x][i])) return true;
    }
    if(x != 1 && !t[x][0])
    {
        int y = fail[x];
        while(!t[y][0]) y = fail[y];
        y = t[y][0];
        if(!val[y] && dfs(y)) return true;
    }

    if(x != 1 && !t[x][1])
    {
        int y = fail[x];
        while(!t[y][1]) y = fail[y];
        y = t[y][1];
        if(!val[y] && dfs(y)) return true;
    }

    inq[x] = 0;
    return false;
}

int main()
{
    sz = 1;
    scanf("%d",&n);
    if(n <= 1) return puts("TAK"),0;
    rep(i,1,n)
    {
        scanf("%s",s + 1);
        insert(s);
    }

    build_ac();

    pre(1,0);
    puts(dfs(1)?"TAK" : "NIE");

    return 0;
}

你可能感兴趣的:(BZOJ,BZOJ刷题录)