CSU2004：Finding words（含指定不相交前后缀的模式串计数)

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int M=1e6+5;
const int N=1e5+5;
const int maxn=27;
int ans[N];
struct ac{
    int trie[M][maxn],end[M],fail[M],Len[M];
    int root,tot;
    int newnode(){
        for(int i=0;i<maxn;i++){
            trie[tot][i]=-1;
        }
        end[tot++]=0;
        return tot-1;
    }
    void init(){
        tot=0;
        root=newnode();
    }
    void insert(string buf,int id){
        int len=buf.size();
        int now=root;
        for(int i=0;i<len;i++){
            if(trie[now][buf[i]-‘a‘]==-1)
                trie[now][buf[i]-‘a‘]=newnode();
            now=trie[now][buf[i]-‘a‘];
        }
        end[now]=id;
        Len[now]=len-1;
    }
    void getfail(){
        queue<int>que;
        while(!que.empty()){
            que.pop();
        }
        fail[root]=root;
        for(int i=0;i<maxn;i++){
            if(trie[root][i]==-1)
                trie[root][i]=root;
            else{
                fail[trie[root][i]]=root;
                que.push(trie[root][i]);
            }
        }
        while(!que.empty()){
            int now=que.front();
            que.pop();
            for(int i=0;i<maxn;i++){
                if(trie[now][i]!=-1){
                    fail[trie[now][i]]=trie[fail[now]][i];
                    que.push(trie[now][i]);
                }
                else
                    trie[now][i]=trie[fail[now]][i];
            }
        }
    }
    void query(string buf){
        int len=buf.size();
        int now=root;
        int flag=(len-1)/2;
        for(int i=0;i<len;i++){
            now=trie[now][buf[i]-‘a‘];
            int tmp=now;
            while(tmp!=root){
                if(Len[tmp]&&end[tmp]!=0&&flag>=Len[tmp])
                    ans[end[tmp]]++;
                tmp=fail[tmp];
            }
        }
    }
}AC;
char sign=‘z‘+1;
string s[M];
char buf[33];
int main(){

    AC.init();
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",buf);
        s[i]=buf;
        s[i]=s[i]+sign+s[i];
    }
    int m;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",buf);
        string s2=buf;
        if(s2=="*"){
            ans[i]=n;
            continue;
        }
        int j; 
        for(j=0;j<s2.size();j++){
            if(s2[j]==‘*‘)
                break;
        }
        s2=s2.substr(j+1,s2.size()-j-1)+sign+s2.substr(0,j);
        AC.insert(s2,i);
    }
    AC.getfail();
    //cout<<"!!"<<endl;
    for(int i=1;i<=n;i++){
        AC.query(s[i]); 
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}