P5231 [JSOI2012]玄武密码

P5231 [JSOI2012]玄武密码

\(AC\) 自动机裸题。

Sol：

• 1.建一颗 \(Trie\) 树。

• 2.处理 \(fail\) 指针。

• 3.文本串跑一遍自动机，把子串都打上标记。

• 4.每个模式串跑一遍 \(Trie\) 树，找到最深标记节点。

Code：

/*
Knowledge : Rubbish Algorithm
Work by :Gym_nastics
Time : O(AC)
*/
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
const int Mod=1e9+7;
const int N=1e6+6;

int read() {
    int x=0,f=0;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) f|=(ch=='-');
    for(;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+(ch&15);
    return f?-x:x;
}

void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>9) print(x/10);
    putchar(x%10+48);
}

int Trie[N][5],fail[N],tot,n,m;
queue<int>q;bool vis[N];

struct Automaton{
    int turn(char ch){
        if(ch=='E') return 1;
        if(ch=='S') return 2;
        if(ch=='W') return 3;
        if(ch=='N') return 4;
    }
    
    void Insert(char *s){
        int pos=0,len=strlen(s);
        for(int i=0;i<len;i++){
            int u=turn(s[i]);
            if(!Trie[pos][u]) Trie[pos][u]=++tot;
            pos=Trie[pos][u];
        }
    }
    
    void Fail(){
        for(int i=1;i<=4;i++) if(Trie[0][i]) q.push(Trie[0][i]);
        while(!q.empty()){
            int u=q.front();q.pop();
            for(int i=1;i<=4;i++){
                if(Trie[u][i]) fail[Trie[u][i]]=Trie[fail[u]][i],q.push(Trie[u][i]);
                else Trie[u][i]=Trie[fail[u]][i];
            }
        }
    }
    
    void arrive(char *s){
        int pos=0,len=strlen(s);
        for(int i=0;i<len;i++){
            pos=Trie[pos][turn(s[i])];
            for(int j=pos;j;j=fail[j])
            if(!vis[j]) vis[j]=true;
            else break;
        }
    }
    
    int Query(char *s){
        int pos=0,res=0,len=strlen(s);
        for(int i=0;i<len;i++){
            pos=Trie[pos][turn(s[i])];
            if(vis[pos]) res=i+1;
        }return res;
    }  
}AC;

char s[N][105],t[N];

signed main() {
   n=read();m=read();scanf("%s",t);
   for(int i=1;i<=m;i++) scanf("%s",s[i]),AC.Insert(s[i]);
   AC.Fail();AC.arrive(t);for(int i=1;i<=m;i++) print(AC.Query(s[i])),putchar('\n');
   return 0;
}