[JSOI 2012] 玄武密码

[题目链接]

https://www.lydsy.com/JudgeOnline/problem.php?id=4327

[算法]

AC自动机
[代码]

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e7 + ;
const int MAXM = 1e5 + ;
const int MAXLEN = ;

int n,m;
int ans[MAXM];
char P[MAXN];
char s[MAXM][MAXLEN];

inline int get_value(char a)
{
        if (a == 'E') return ;
        if (a == 'S') return ;
        if (a == 'W') return ;
        if (a == 'N') return ;
}
struct AC_Automation
{
        int tot;
        int root;
        struct Node
        {
                int child[];
                int fail;
                bool visited;
        } trie[MAXN];
        inline void insert(char *s)
        {
                int now = root;
                int len = strlen(s + );
                for (int i = ; i <= len; i++)
                {
                        int val = get_value(s[i]);
                        if (!trie[now].child[val]) trie[now].child[val] = ++tot;
                        now = trie[now].child[val];
                }
        }
        inline void rebuild()
        {
                int l,r;
                static int q[MAXN];
                q[l = r = ] = root;
                trie[root].fail = -;
                while (l <= r)
                {
                        int cur = q[l];
                        l++;
                        for (int i = ; i < ; i++)
                        {
                                if (trie[cur].child[i])
                                {
                                        if (cur == root)
                                                trie[trie[cur].child[i]].fail = ;
                                        else
                                        {
                                                int p = trie[cur].fail;
                                                while (p != -)
                                                {
                                                        if (trie[p].child[i])
                                                        {
                                                                trie[trie[cur].child[i]].fail = trie[p].child[i];
                                                                break;
                                                        } else p = trie[p].fail;
                                                }
                                        }
                                        q[++r] = trie[cur].child[i];
                                 }
                        }
                }
        }
        inline void getans(char *s)
        {
                int now = root;
                int len = strlen(s + );
                for (int i = ; i <= len; i++)
                {
                        int val = get_value(s[i]);
                        int p = now;
                        while (p != -)
                        {
                                if (trie[p].child[val]) break;
                                else p = trie[p].fail;
                        }
                        if (p == -)
                        {
                                now = ;
                                continue;
                        } else now = p = trie[p].child[val];
                        while (p)
                        {
                                if (!trie[p].visited)
                                {
                                        trie[p].visited = true;
                                        p = trie[p].fail;
                                } else break;
                        }
                }
        }
        inline void calc(int pos,char *s)
        {
                int now = root;
                int len = strlen(s + );
                for (int i = ; i <= len; i++)
                {
                        int val = get_value(s[i]);
                        now = trie[now].child[val];
                        if (trie[now].visited) ans[pos] = i;
                }
        }
} ACAM;

int main()
{

        scanf("%d%d",&n,&m);
        scanf("%s",P + );
        for (int i = ; i <= m; i++) scanf("%s",s[i] + );
        for (int i = ; i <= m; i++) ACAM.insert(s[i]);
        ACAM.rebuild();
        ACAM.getans(P);
        for (int i = ; i <= m; i++) ACAM.calc(i,s[i]);
        for (int i = ; i <= m; i++) printf("%d\n",ans[i]);

        return ;

}