BZOJ 1355: [Baltic2009]Radio Transmission AC自动机/KMP

BZOJ 1355: [Baltic2009]Radio Transmission AC自动机/KMP

AC自动机

Code:

// luogu-judger-enable-o2
#include <bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 1000001 
#define idx (p[i]-'a')
using namespace std;
char str[maxn]; 
queue<int>Q; 
int ch[maxn][28], root,cnt,map1[maxn],map2[maxn],l,f[maxn]; 
void ins(char p[]){
    int len=strlen(p),x=root; 
    for(int i=0;i<len;++i) {  
        if(!ch[x][idx]) ch[x][idx]=++cnt;
        x=ch[x][idx]; 
        map1[i+1]=x, map2[x]=i+1; 
    }
}
void getfail(){
    for(int i=0;i<28;++i) if(ch[root][i]) Q.push(ch[root][i]); 
    while(!Q.empty()){
        int u=Q.front();Q.pop(); 
        for(int i=0;i<28;++i) {
            int q=ch[u][i]; 
            if(!q) { ch[u][i]=ch[f[u]][i]; continue;}
            Q.push(q),f[q]=ch[f[u]][i]; 
        }
    }
}
int main()
{
    //setIO("input");
    scanf("%d%s",&l,str);
    ins(str),getfail();
    printf("%d",l-map2[f[map1[l]]]); 
    return 0; 
}

　　

posted @ 2019-05-29 15:01 EM-LGH 阅读( ...) 评论( ...) 编辑 收藏