AC自动机

不是很懂AC自动机,只能照着别人写的写了个,实在是无语,慢慢学学AC自动机吧,希望自己能懂,并且自己能敲出来,留下做个标记吧。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

char str[1000005];
int head, tail;

struct node{
   node *next[26];
   node *fail;
   int count;
   node(){
      count = 0;
      fail = NULL;
 //     memset(next, NULL, sizeof(next));
      for(int i = 0; i < 26; ++i)
        next[i] = NULL;
   }
}*q[500001];

void insert(node *root){
   node *p = root;
   int i= 0, index;
   while(str[i]){
       index = str[i]-'a';
       if(p->next[index] == NULL) p->next[index] = new node();
       p = p->next[index];
       i++;
   }
   p->count++;
}

void build_ac(node *root)
{
    root->fail = NULL;
    q[head++] = root;
    while(head != tail)
    {
       node *temp = q[tail++];
       node *p = NULL;
       
       for(int i = 0; i < 26; ++i){
          if(temp->next[i] != NULL){
             if(temp==root) temp->next[i]->fail=root;
             else {
                  p = temp->fail;
                  while(p != NULL)
                  {
                      if(p->next[i]!=NULL)
                      {
                          temp->next[i]->fail = p->next[i];   
                          break;                 
                      }                 
                      p = p->fail;             
                  } 
                  if(p == NULL) temp->next[i]->fail=root;
             }
             q[head++] = temp->next[i];
          }    
       }               
    }     
}

int query(node *root)
{
    int i=0, cnt = 0, index;
    node *p = root;
    
    while(str[i])
    {
       index = str[i]-'a';
       while(p->next[index] == NULL && p!=root) p=p->fail;
       p = p->next[index];
       if(p == NULL) p = root;
       
       node *temp = p;
       
       while(temp!=root && temp->count!=-1)
       {
           cnt+=temp->count;
           temp->count = -1;
           temp=temp->fail;                 
       }
       i++;                 
    } 
    return cnt;
}


int main()
{
    int n, t;
    scanf("%d", &t);
    
    while(t--)
    {
        head = tail = 0;
        node *root = new node();
        scanf("%d", &n);
        while(n--)
        {
            scanf("%s", str);
            insert(root);       
        }      
        build_ac(root);
        scanf("%s", str);
        printf("%d\n", query(root));
              
    }
    return 0;
}
   
   


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值