hdu2222 AC自动机

hdu2222:

题意:给出多个字符串(模式串),再给出一个待匹配串,问模式串中出现多少待匹配串

首先,一般的坑点:①待模式串可以相同,但是算多个模式串,需要重复计算 ②待匹配串中可以重复出现多次同一个模式串,但是这些重复出现不计入总匹配串数

但事实上,我一开始就并没有被这些迷惑,只是我使用的AC自动机模板并不是非常先进```中间有一处错误,是将所有以字典树上某串为原串或该串的后缀串为原串的总个数都累加到该串上,然后计算的时候加上该串的值即可。但是我wa了一发后很快发现此时只清除了该串的结尾个数,后面再匹配到其后缀串又会有重复计数的情况,然后我在加和之后再清除其所有后缀串的个数,但是仍然是wa,虽然并不清楚到底此处为什么wa,但是改为建树建nxt和fail时不加和,而匹配到时循环回去加就AC了```

果然太弱了连怎么WA的都不知道昂```

 1 #include<stdio.h>

 2 #include<string.h>

 3 #include<queue>

 4 using namespace std;

 5 const int maxm=500006;

 6 

 7 char s[1000010],word[55];

 8 int nxt[maxm][26],cnt[maxm],fail[maxm],size;

 9 

10 int newnode(){

11     memset(nxt[size],0,sizeof(nxt[size]));

12     fail[size]=cnt[size]=0;

13     return size++;

14 }

15 

16 void insert(char s[]){

17     int i,p=0;

18     for(i=0;s[i];i++){

19         int &x=nxt[p][s[i]-'a'];

20         p=x?x:x=newnode();

21     }

22     cnt[p]++;

23 }

24 

25 void makenxt(){

26     int i;

27     queue<int>q;

28     q.push(0);

29     while(!q.empty()){

30         int u=q.front();

31         q.pop();

32         for(i=0;i<26;i++){

33             int v=nxt[u][i];

34             if(v==0)nxt[u][i]=nxt[fail[u]][i];

35             else q.push(v);    

36             if(u&&v){

37                 fail[v]=nxt[fail[u]][i];

38             }

39         }

40     }

41 }

42 

43 int query(char s[]){

44     int ans=0,d=0;

45     for(int i=0;s[i];i++){

46         d=nxt[d][s[i]-'a'];

47         int tmp=d;

48         while(tmp!=0){

49             ans+=cnt[tmp];

50             cnt[tmp]=0;

51             tmp=fail[tmp];

52         }

53     }

54     return ans;

55 }

56 

57 int main(){

58     int T;

59     scanf("%d",&T);

60     while(T--){

61         int n;

62         size=0,newnode();

63         scanf("%d",&n);

64         for(int i=1;i<=n;i++){

65             scanf("%s",word);

66             insert(word);

67         }

68         makenxt();

69         scanf("%s",s);

70         printf("%d\n",query(s));

71     }

72     return 0;

73 }
hdu2222

 

你可能感兴趣的:(AC自动机)