[hdu1277]全文检索(AC自动机)

解题关键：AC自动机模板题，注意字符匹配时若无法匹配，直接用%s即可。

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int N=;
 const int MAXN=;
 int num,ans[],nn;
 bool vis[];
 bool flag=false;
 struct Trie{//数组形式
     int Next[MAXN][N],Fail[MAXN],End[MAXN],root,tot;//大小为所以匹配字符串的总和
     int newnode(){//结构体内部用
         for(int i=;i<N;i++) Next[tot][i]=-;
         End[tot++]=;
         return tot-;
     }
     void init(){
         tot=;
         root=newnode();
     }
     void insert(char buf[],int x){
         int len=strlen(buf);
         int now=root;//now是temp指针
         for(int i=;i<len;i++){
             int k=buf[i]-'0';
             if(Next[now][k]==-)  Next[now][k]=newnode();//next数组代表的是下一个字符索引
             now=Next[now][k];
         }
         End[now]=x;//end数组是当前字符串的个数.字典中可能有相同的单词，若只算一次，改为1.
     }
     void build(){//构造fail指针，后缀是某些前缀
         queue<int>que;
         Fail[root]=root;
         for(int i=;i<N;i++){
             if(Next[root][i]==-) Next[root][i]=root;
             else{
                 Fail[Next[root][i]]=root;
                 que.push(Next[root][i]);
             }
         }
         while(!que.empty()){//bfs,会将所有的匹配子串都遍历到
             int now=que.front();
             que.pop();
             for(int i=;i<N;i++){
                 if(Next[now][i]==-) Next[now][i]=Next[Fail[now]][i];
                 else{
                     Fail[Next[now][i]]=Next[Fail[now]][i];//fail指向最长的
                     que.push(Next[now][i]);
                 }
             }
         }
     }
     void query(char buf[]){
         int len=strlen(buf),now=root;
         for(int i=;i<len;i++){
             now=Next[now][buf[i]-'0'];
             int temp=now;
             while(temp!=root){
                 if(End[temp]&&!vis[temp]) ans[nn++]=End[temp],vis[temp]=true,flag=true;
                 temp=Fail[temp];
             }
         }
     }
 };

 Trie ac;
 char buf[],buf2[],tmp[];
 int n,m;
 int main(){
     while(scanf("%d%d",&m,&n)!=EOF){
         memset(ans,,sizeof ans);
         memset(vis,,sizeof vis);
         nn=;
         ac.init();
         int t=;
         flag=false;
         for(int i=;i<=m;i++){
             scanf("%s",tmp);
             strcat(buf,tmp);
         }
            for(int i=;i<=n;i++){
                scanf("%*s %*s %d] %s",&num,buf2);
                ac.insert(buf2,num);
         }
            ac.build();//不要忘记build
         ac.query(buf);
         if(flag){
             printf("Found key:");
             for(int i=;i<nn;i++){
                 printf(" [Key No. %d]",ans[i]);
             }
             printf("\n");
         }
         else printf("No key can be found !\n");
     }
     return ;
 }