[hdu1277]全文检索(AC自动机)
解题关键：AC自动机模板题，注意字符匹配时若无法匹配，直接用%s即可。
1 #include<bits/stdc++.h> 2 using namespace std; 3 typedef long long ll; 4 const int N=12; 5 const int MAXN=600010; 6 int num,ans[10020],nn; 7 bool vis[600010]; 8 bool flag=false; 9 struct Trie{//数组形式 10 int Next[MAXN][N],Fail[MAXN],End[MAXN],root,tot;//大小为所以匹配字符串的总和 11 int newnode(){//结构体内部用 12 for(int i=0;i<N;i++) Next[tot][i]=-1; 13 End[tot++]=0; 14 return tot-1; 15 } 16 void init(){ 17 tot=0; 18 root=newnode(); 19 } 20 void insert(char buf[],int x){ 21 int len=strlen(buf); 22 int now=root;//now是temp指针 23 for(int i=0;i<len;i++){ 24 int k=buf[i]-'0'; 25 if(Next[now][k]==-1) Next[now][k]=newnode();//next数组代表的是下一个字符索引 26 now=Next[now][k]; 27 } 28 End[now]=x;//end数组是当前字符串的个数.字典中可能有相同的单词，若只算一次，改为1. 29 } 30 void build(){//构造fail指针，后缀是某些前缀 31 queue<int>que; 32 Fail[root]=root; 33 for(int i=0;i<N;i++){ 34 if(Next[root][i]==-1) Next[root][i]=root; 35 else{ 36 Fail[Next[root][i]]=root; 37 que.push(Next[root][i]); 38 } 39 } 40 while(!que.empty()){//bfs,会将所有的匹配子串都遍历到 41 int now=que.front(); 42 que.pop(); 43 for(int i=0;i<N;i++){ 44 if(Next[now][i]==-1) Next[now][i]=Next[Fail[now]][i]; 45 else{ 46 Fail[Next[now][i]]=Next[Fail[now]][i];//fail指向最长的 47 que.push(Next[now][i]); 48 } 49 } 50 } 51 } 52 void query(char buf[]){ 53 int len=strlen(buf),now=root; 54 for(int i=0;i<len;i++){ 55 now=Next[now][buf[i]-'0']; 56 int temp=now; 57 while(temp!=root){ 58 if(End[temp]&&!vis[temp]) ans[nn++]=End[temp],vis[temp]=true,flag=true; 59 temp=Fail[temp]; 60 } 61 } 62 } 63 }; 64 65 Trie ac; 66 char buf[60004],buf2[60004],tmp[6002]; 67 int n,m; 68 int main(){ 69 while(scanf("%d%d",&m,&n)!=EOF){ 70 memset(ans,0,sizeof ans); 71 memset(vis,0,sizeof vis); 72 nn=0; 73 ac.init(); 74 int t=0; 75 flag=false; 76 for(int i=1;i<=m;i++){ 77 scanf("%s",tmp); 78 strcat(buf,tmp); 79 } 80 for(int i=1;i<=n;i++){ 81 scanf("%*s %*s %d] %s",&num,buf2); 82 ac.insert(buf2,num); 83 } 84 ac.build();//不要忘记build 85 ac.query(buf); 86 if(flag){ 87 printf("Found key:"); 88 for(int i=0;i<nn;i++){ 89 printf(" [Key No. %d]",ans[i]); 90 } 91 printf("\n"); 92 } 93 else printf("No key can be found !\n"); 94 } 95 return 0; 96 }
