bzoj4327: JSOI2012 玄武密码 AC自动机

bzoj4327: JSOI2012 玄武密码

Description

在美丽的玄武湖畔,鸡鸣寺边,鸡笼山前,有一块富饶而秀美的土地,人们唤作进香河。相传一日,一缕紫气从天而至,只一瞬间便消失在了进香河中。老人们说,这是玄武神灵将天书藏匿在此。
很多年后,人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是,这份带有玄武密码的文字,与玄武湖南岸台城的结构有微妙的关联。于是,漫长的破译工作开始了。
经过分析,我们可以用东南西北四个方向来描述台城城砖的摆放,不妨用一个长度为N的序列来描述,序列中的元素分别是‘E’,‘S’,‘W’,‘N’,代表了东南西北四向,我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象,分别是东之青龙,西之白虎,南之朱雀,北之玄武,对东南西北四向相对应。
现在,考古工作者遇到了一个难题。对于每一段文字,其前缀在母串上的最大匹配长度是多少呢?

Input

第一行有两个整数,N和M,分别表示母串的长度和文字段的个数。
第二行是一个长度为N的字符串,所有字符都满足是E,S,W和N中的一个。
之后M行,每行有一个字符串,描述了一段带有玄武密码的文字。依然满足,所有字符都满足是E,S,W和N中的一个。

Output

输出有M行,对应M段文字。
每一行输出一个数,表示这一段文字的前缀与母串的最大匹配串长度。

Sample Input

7 3
SNNSSNS
NNSS
NNN
WSEE

Sample Output

4
2
0

HINT

对于100%的数据,N<=10^7,M<=10^5,每一段文字的长度<=100。
应上传者要求,此题不公开,如有异议,请提出.

分析

算是AC自动机的一个套路吧。
从模板串入手比较难,所以从匹配串入手。
对匹配串建立AC自动机。
然后模板串上去跑,跑到的所有节点标记。
然后标记通过fail-tree传导。
对于每个串再计算一下最长的长度即可。
优秀的线性算法。

代码

#include<cstdio>
const int N = 1e7 + 10, M = 1e5 + 10;
int S[N], q[N], f[N], ch[N][4], s[M][110], t[110], sz, n, m, L, R;
bool vis[N];
void Ins(int &p, int c) {p = ch[p][c] ? ch[p][c] : ch[p][c] = ++sz;}
void rc(int *s, bool p) {
    char ch = getchar(); int u = 0;
    for(;ch < 'A' || ch > 'Z'; ch = getchar()) ;
    for(;ch >= 'A' && ch <= 'Z'; ch = getchar()) {s[++s[0]] = t[ch]; if(p) Ins(u, t[ch]);}
}
void Build() {
    L = R = 0;
    for(int c = 0;c < 4; ++c) if(ch[0][c]) q[++R] = ch[0][c];
    for(int u = q[++L]; L <= R; u = q[++L])
        for(int c = 0; c < 4; ++c)
            if(!ch[u][c]) ch[u][c] = ch[f[u]][c];
            else f[q[++R] = ch[u][c]] = ch[f[u]][c];
}
int Que(int *s) {
    int r = 0, u = 0; for(int i = 1;i <= s[0]; u = ch[u][s[i++]]) if(vis[u]) r = i - 1;
    return vis[u] ? s[0] : r;
}
int main() {
    t['E'] = 0; t['S'] = 1; t['W'] = 2; t['N'] = 3;
    scanf("%d%d", &n, &m); rc(S, 0);
    for(int i = 1;i <= m; ++i) rc(s[i], 1);
    Build(); int u = 0; for(int i = 1;i <= n; u = ch[u][S[i++]]) vis[u] = true; vis[u] = true;
    for(int i = R; i; --i) vis[f[q[i]]] |= vis[q[i]];
    for(int i = 1;i <= m; ++i) printf("%d\n", Que(s[i]));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/lvzelong2014/article/details/80527106
今日推荐