BZOJ 4327: JSOI2012 玄武密码 后缀自动机

BZOJ 4327: JSOI2012 玄武密码 后缀自动机

Code:

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 18000001
using namespace std;
char str[maxn];  
int idx(char i)
{
    if(i=='S') return 0;
    if(i=='N') return 1; 
    if(i=='W') return 2; 
    if(i=='E') return 3; 
}
namespace SAM
{
    int ch[maxn][4],f[maxn],dis[maxn]; 
    int tot,last;
    void init()  
    { 
        tot=last=1;  
    }
    void ins(int c)
    {
        int np=++tot,p=last;   
        last=np,dis[np]=dis[p]+1; 
        while(p&&!ch[p][c]) ch[p][c]=np,p=f[p]; 
        if(!p) f[np]=1;                      
        else     
        {
            int q=ch[p][c];
            if(dis[q]==dis[p]+1) f[np]=q;
            else 
            {     
                int nq=++tot;
                dis[nq]=dis[p]+1;     
                memcpy(ch[nq],ch[q],sizeof(ch[q])); 
                f[nq]=f[q],f[q]=f[np]=nq;                  
                while(p&&ch[p][c]==q) ch[p][c]=nq,p=f[p];      
            }
        }
    } 
}; 
int main()
{
    // setIO("input"); 
    int n,m;
    scanf("%d%d%s",&n,&m,str+1);
    SAM::init();  
    for(int i=1;i<=n;++i) SAM::ins(idx(str[i]));     
    while(m--)
    { 
        scanf("%s",str+1);
        int k=strlen(str+1),p=1,cur=0;
        for(int i=1;i<=k;++i)
        {
            int c=idx(str[i]); 
            if(!SAM::ch[p][c]) 
                break; 
            else 
                ++cur,p=SAM::ch[p][c]; 
        }
        printf("%d\n",cur); 
    }
    return 0; 
}

　　

posted @ 2019-06-10 18:27 EM-LGH 阅读( ...) 评论( ...) 编辑 收藏