AC自动机模板+经典例题

Posted 2021-03-15 pengge666

AC自动机模板

经典例题

Keywords Search HDU - 2222

【求目标串中出现了几个模式串】

【（注意：模式串可能会重复）】

模板：

 1 const int maxn=26;
 2 struct Trie
 3 {
 4     int next[500010][maxn];
 5     int fail[500010],end[500010];
 6     int root,L;
 7     int newnode() //初始化 
 8     {
 9         for(int i=0;i<26;i++)
10         next[L][i]=-1;
11         end[L++]=0;
12         return L-1;
13     }
14 
15     void init()//初始化 
16     {
17         L=0;
18         root=newnode();
19     }
20     void insert(char s[])//构造树trie 
21     {
22         int len=strlen(s);
23         int now=root;
24         for(int i=0;i<len;i++)
25         {
26             if(next[now][s[i]-‘a‘]==-1)
27               next[now][s[i]-‘a‘]=newnode();
28             now=next[now][s[i]-‘a‘];  
29         }
30         end[now]++;
31     }
32 
33     void build()//构造失败指针 
34     {
35         queue<int>Q;
36         fail[root]=root;
37         for(int i=0;i<26;i++)
38         if(next[root][i]==-1)
39             next[root][i]=root;
40         else
41         {
42             fail[next[root][i]]=root;
43             Q.push(next[root][i]);
44         }
45         while(!Q.empty())
46         {
47             int now=Q.front();
48             Q.pop();
49             for(int i=0;i<26;i++)
50                 if(next[now][i]==-1)
51                     next[now][i]=next[fail[now]][i];
52                     else
53                     {
54                         fail[next[now][i]]=next[fail[now]][i];
55                         Q.push(next[now][i]);
56                     }
57 
58                 }       
59     }
60 
61   void query(char buf[])//查询 
62   {
63     int len=strlen(buf);
64     int now=root;
65     int sum=0;
66     for(int i=0;i<len;i++)
67     {
68             now = next[now][buf[i]-‘a‘];
69             int temp = now;
70             while(temp != root)
71             {
72 
73                 sum+=end[temp];
74                 end[temp]=0;//避免重复计算 
75                 temp = fail[temp];
76             }
77       }
78         printf("%d
",sum);
79     }
80   };

病毒侵袭 HDU - 2896  

【统计目标串中模式串的个数】

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<queue>
  6 #include<vector>
  7 #include<cmath>
  8 #include<string>
  9 #include<map>
 10 #define LL long long  
 11 using namespace std;
 12 const int maxn=128;//模板部分
 13 struct Trie
 14 {
 15     int next[210*500][maxn];
 16     int fail[210*500],end[210*500];
 17     int root,L;
 18     int newnode() //初始化 
 19     {
 20         for(int i=0;i<128;i++)
 21         next[L][i]=-1;
 22         end[L++]=-1;
 23         return L-1;
 24     }
 25 
 26     void init()//初始化 
 27     {
 28         L=0;
 29         root=newnode();
 30     }
 31     void insert(char s[],int id)//构造树trie 
 32     {
 33         int len=strlen(s);
 34         int now=root;
 35         for(int i=0;i<len;i++)
 36         {
 37             if(next[now][s[i]]==-1)
 38               next[now][s[i]]=newnode();
 39             now=next[now][s[i]];  
 40         }
 41         end[now]=id;//记录单词的id
 42     }
 43 
 44     void build()//构造失败指针 
 45     {
 46         queue<int>Q;
 47         fail[root]=root;
 48         for(int i=0;i<128;i++)
 49         if(next[root][i]==-1)
 50             next[root][i]=root;
 51         else
 52         {
 53             fail[next[root][i]]=root;
 54             Q.push(next[root][i]);
 55         }
 56         while(!Q.empty())
 57         {
 58             int now=Q.front();
 59             Q.pop();
 60             for(int i=0;i<128;i++)
 61                 if(next[now][i]==-1)
 62                     next[now][i]=next[fail[now]][i];
 63                     else
 64                     {
 65                         fail[next[now][i]]=next[fail[now]][i];
 66                         Q.push(next[now][i]);
 67                     }
 68 
 69                 }       
 70     }
 71 
 72 
 73   bool vis[510];
 74   bool query(char buf[],int n,int id)//查询 
 75   {
 76     int len=strlen(buf);
 77     int now=root;
 78     memset(vis,0,sizeof(vis));
 79     bool flag=false;
 80     for(int i=0;i<len;i++)
 81     {
 82             now = next[now][buf[i]];
 83             int temp = now;
 84             while(temp != root)
 85             {
 86                 if(end[temp] != -1)
 87                 {
 88                     vis[end[temp]] = true;
 89                     flag = true;
 90                 }
 91                 temp = fail[temp];
 92             }
 93       }
 94       if(!flag)return false;
 95         printf("web %d:",id);
 96         for(int i = 1;i <= n;i++)
 97             if(vis[i])
 98                 printf(" %d",i);
 99         printf("
");
100         return true;
101     }
102   };
103 char str1[205]; 
104 char str2[10010];
105 Trie ac;
106 int main()
107 {
108     int n,m;
109     while(scanf("%d",&n) != EOF)
110     {
111         ac.init();
112         for(int i = 1;i <= n;i++)
113         {
114             scanf("%s",str1);
115             ac.insert(str1,i);
116         }
117         ac.build();
118         int ans = 0;
119         scanf("%d",&m);
120         for(int i = 1;i <= m;i++)
121         {
122             scanf("%s",str2);
123             if(ac.query(str2,n,i))
124                 ans++;
125         }
126         printf("total: %d
",ans);
127     }
128     return 0;
129 }

 

病毒侵袭持续中 HDU - 3065

【思路：和病毒侵袭 HDU - 2896 这题差不多，只是多了一个计数。】

 

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<queue>
  6 #include<vector>
  7 #include<cmath>
  8 #include<string>
  9 #include<map>
 10 
 11 using namespace std;
 12 #define int long long 
 13 const int maxn=128;
 14 
 15 char str1[1010][100]; 
 16 char str2[2000005];
 17 
 18 char STR[1003][66];
 19 char SSS[2000100];
 20 struct Trie{
 21     int next[2010*50][maxn];
 22     int fail[2010*50],end[2010*50];
 23     int root,L;
 24     int newnode() //初始化 
 25     {
 26         for(int i=0;i<128;i++)
 27         next[L][i]=-1;
 28         end[L++]=-1;
 29         return L-1;
 30     }
 31 
 32     void init()//初始化 
 33     {
 34         L=0;
 35         root=newnode();
 36     }
 37     void insert(char s[],int id)//构造树trie 
 38     {
 39         int len=strlen(s);
 40         int now=root;
 41         for(int i=0;i<len;i++)
 42         {
 43             if(next[now][s[i]]==-1)
 44               next[now][s[i]]=newnode();
 45             now=next[now][s[i]];  
 46         }
 47         end[now]=id;//记录单词的id
 48     }
 49 
 50     void build()//构造失败指针 
 51     {
 52         queue<int>Q;
 53         fail[root]=root;
 54         for(int i=0;i<128;i++)
 55         if(next[root][i]==-1)
 56             next[root][i]=root;
 57         else
 58         {
 59             fail[next[root][i]]=root;
 60             Q.push(next[root][i]);
 61         }
 62         while(!Q.empty())
 63         {
 64             int now=Q.front();
 65             Q.pop();
 66             for(int i=0;i<128;i++)
 67                 if(next[now][i]==-1)
 68                     next[now][i]=next[fail[now]][i];
 69                     else
 70                     {
 71                         fail[next[now][i]]=next[fail[now]][i];
 72                         Q.push(next[now][i]);
 73                     }
 74 
 75                 }       
 76     }
 77 
 78 
 79   int vis[1666];//记录出现的次数
 80   void  query(char buf[],int n)//查询 
 81   {
 82     int len=strlen(buf);
 83     int now=root;
 84     memset(vis,0,sizeof(vis));
 85     bool flag=false;
 86     for(int i=0;i<len;i++)
 87     {
 88             now = next[now][buf[i]];
 89             int temp = now;
 90             while(temp != root)
 91             {
 92                 if(end[temp] != -1)
 93                 {
 94                     vis[end[temp]]++;
 95                     flag = true;
 96                 }
 97                 temp = fail[temp];
 98             }
 99       }
100       if(flag ){
101           for(int i=1;i<=n;i++){
102           if(vis[i]){
103               printf("%s: %lld
",STR[i],vis[i]);    
104         }
105       }
106       }
107       return ;
108       
109       
110     }
111 }AC;
112 
113 signed  main(){
114     int n;
115     
116     while(~scanf("%lld",&n)){
117      
118     AC.init();
119     for(int i=1;i<=n;i++){
120         scanf("%s",STR[i]);
121         AC.insert(STR[i],i);
122     }
123     AC.build();
124     scanf("%s",SSS);
125     AC.query(SSS,n); 
126     } 
127     return 0; 
128 }