【题解】Tree-String Problem Codeforces 291E AC自动机

Prelude

传送到Codeforces:(/ω\)……… (/ω•\)


Solution

很水的一道题。
对查询的串建出来AC自动机,然后树上随便跑跑就行了。
为什么要写这篇题解呢?
我第一眼看到这个题:“哈哈,有根树上的路径信息查询,点分治就好了,被我秒啦!”
“这个题好像是某Qualification Round的题啊。。。怎么Qual就出点分治啊,真毒瘤。”
然后码码码。。。
“怎么TLE了,卡卡常。”
卡常ing。。。
“怎么又WA了,难道卡哈希?毒瘤毒瘤。”
debugging。。。
“算了调不出来了拿tourist的代码拍一下吧。”
“这场比赛怎么AK了三百个人啊?这题怎么代码这么短啊?”
“。。。似乎KMP然后dfs一下就没了。。。我简直药丸。。。”
然后事实证明KMP的复杂度是错的,很容易就卡掉了,卡的方法也很简单,大家自己想想吧。。。
如果用KMP的话是TLE on test 30,如果是比赛就直接FST了。。。
所以用AC自动机就没问题了喵~


Code

#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>

using namespace std;
const int MAXN = 400010;
int _w;

int n, ch[MAXN], m;
char str[MAXN];

namespace G {
    int head[MAXN], nxt[MAXN], to[MAXN], eid;
    void init() {
        eid = 0;
        memset(head, -1, sizeof head);
    }
    void adde( int u, int v ) {
        to[eid] = v, nxt[eid] = head[u], head[u] = eid++;
    }
}

void add_edge( int u, int v, char *str ) {
    for( char *p = str; *p; ++p ) {
        ch[++m] = *p - 'a';
        if( p == str ) G::adde(u, m);
        else G::adde(m-1, m);
    }
    G::adde(m, v);
}

namespace AC {
    int ch[MAXN][26], nid;
    bool word[MAXN];
    queue<int> q;
    int f[MAXN];
    
    void insert( char *s ) {
        int u = 0;
        for( ; *s; ++s ) {
            int c = *s - 'a';
            if( !ch[u][c] )
                ch[u][c] = ++nid;
            u = ch[u][c];
        }
        word[u] = 1;
    }
    void build() {
        for( int i = 0; i < 26; ++i )
            if( ch[0][i] )
                q.push( ch[0][i] );
        while( !q.empty() ) {
            int u = q.front(); q.pop();
            for( int i = 0; i < 26; ++i ) {
                int v = ch[u][i];
                if( !v ) {
                    ch[u][i] = ch[f[u]][i];
                    continue;
                }
                int w = f[u];
                while( w && !ch[w][i] ) w = f[w];
                f[v] = ch[w][i];
                q.push(v);
            }
        }
    }
}

int ans;
void solve( int u, int o ) {
    using namespace G;
    if( ch[u] != -1 ) {
        o = AC::ch[o][ch[u]];
        if( AC::word[o] ) ++ans;
    }
    for( int i = head[u]; ~i; i = nxt[i] )
        solve( to[i], o );
}

int main() {
    _w = scanf( "%d", &n );
    m = n;
    G::init();
    for( int i = 2; i <= n; ++i ) {
        int pa;
        _w = scanf( "%d%s", &pa, str );
        add_edge(pa, i, str);
    }
    for( int i = 1; i <= n; ++i )
        ch[i] = -1;
    _w = scanf( "%s", str );
    AC::insert(str);
    AC::build();
    solve(1, 0);
    printf( "%d\n", ans );
    return 0;
}
posted @ 2017-12-22 17:02 mlystdcall 阅读(...) 评论(...) 编辑 收藏