首页 > 代码库 > HDU 2896 病毒侵袭 AC自动机
HDU 2896 病毒侵袭 AC自动机
题意:给你n个模式串,m个文本串,问你m个文本串中分别出现了几个模式串
解题思路:AC自动机模板题,我们只需要把 end 变为 n的序号就行。但是要注意这里所有串的范围
解题代码:
1 // File Name: temp.cpp 2 // Author: darkdream 3 // Created Time: 2014年09月11日 星期四 15时18分128秒 4 5 #include<vector> 6 #include<list> 7 #include<map> 8 #include<set> 9 #include<deque> 10 #include<stack> 11 #include<bitset> 12 #include<algorithm> 13 #include<functional> 14 #include<numeric> 15 #include<utility> 16 #include<sstream> 17 #include<iostream> 18 #include<iomanip> 19 #include<cstdio> 20 #include<cmath> 21 #include<cstdlib> 22 #include<cstring> 23 #include<ctime> 24 #include<queue> 25 #define LL long long 26 #define maxn 100010 27 using namespace std; 28 vector<int> ans; 29 struct Trie 30 { 31 int next[maxn][128],fail[maxn],end[maxn]; 32 int root, L; 33 int newnode() 34 { 35 for(int i = 30 ;i < 127;i ++) 36 next[L][i] = -1; 37 end[L++] = 0 ; 38 return L-1; 39 } 40 int init() 41 { 42 int L = 0 ; 43 root = newnode(); 44 } 45 int insert(char buf[],int K) 46 { 47 int len = strlen(buf); 48 int now = root; 49 for(int i = 0 ;i < len ;i ++) 50 { 51 if(next[now][buf[i]] == -1) 52 { 53 next[now][buf[i]] = newnode(); 54 } 55 now = next[now][buf[i]]; 56 } 57 end[now] = K; 58 } 59 void build() 60 { 61 queue<int> Q; 62 //这里为何要用优先队列来解决这个问题,主要原因是fail指针是按层数来的。 63 fail[root] = root; 64 for(int i = 30;i < 127;i ++) 65 { 66 if(next[root][i] == -1) 67 { 68 next[root][i] = root; //指向root 是没有问题的,我们主要是通过end 数组对个数进行计数的。 69 }else{ 70 fail[next[root][i]] = root; 71 Q.push(next[root][i]); 72 } 73 } 74 while(!Q.empty()) 75 { 76 int now = Q.front(); 77 Q.pop(); 78 for(int i = 30 ;i < 127;i ++) 79 { 80 if(next[now][i] == -1) 81 next[now][i] = next[fail[now]][i]; 82 else{ 83 fail[next[now][i]] = next[fail[now]][i]; 84 Q.push(next[now][i]); 85 } 86 } 87 } 88 } 89 void query(char buf[]) 90 { 91 int len = strlen(buf); 92 int now = root ; 93 for(int i = 0 ;i < len;i ++) 94 { 95 now = next[now][buf[i]]; 96 int temp = now ; 97 while(temp != root) 98 { 99 if(end[temp])100 ans.push_back(end[temp]); 101 temp = fail[temp];102 }103 }104 }105 };106 char buf[10010];107 Trie ac;108 int main(){109 int n;110 while(scanf("%d",&n) != EOF)111 {112 ac.init();113 getchar();114 for(int i = 1 ;i <= n;i ++)115 {116 scanf("%[^\n]",buf);117 getchar();118 ac.insert(buf,i);119 }120 ac.build();121 int m;122 scanf("%d",&m);123 getchar();124 int sum = 0 ; 125 for(int i = 1;i <= m;i ++)126 {127 scanf("%s",buf);128 getchar();129 //printf("***");130 ans.clear();131 ac.query(buf);132 if(ans.size() != 0 )133 {134 sum ++;135 }else{136 continue;137 }138 sort(ans.begin(),ans.end());139 printf("web %d: ",i);140 for(int j = 0 ;j < ans.size();j ++)141 printf(j == 0 ?"%d":" %d",ans[j]);142 printf("\n");143 }144 printf("total: %d\n",sum);145 }146 return 0;147 }
HDU 2896 病毒侵袭 AC自动机
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。