码迷,mamicode.com
首页 > 其他好文 > 详细

模板—trie图

时间:2019-06-14 21:25:30      阅读:89      评论:0      收藏:0      [点我收藏+]

标签:自动机   cout   end   using   namespace   include   alt   play   ace   

做了某题之后发现trie的AC自动机太垃圾了,动不动就TLE,然后我就去学了trie图。

技术图片
#include<iostream>
#include<cstdio>
using namespace std;
struct trie
{
    int count;
    trie *fail,*nxt[26];
    trie()
    {
        count=0;
        fail=NULL;
        for(int i=0;i<26;i++)nxt[i]=NULL;
    }
}*q[1000000],*root=new trie();
int head,tail;
char keyword[1000000],str[1000000];

void insert(char s[],trie *root)
{
    int i=0,index;
    trie *p=root;
    while(s[i])
    {
        index=s[i]-a;
        if(p->nxt[index]==NULL)p->nxt[index]=new trie();
        p=p->nxt[index];
        i++;
    }
    p->count++;
}
void build_ac(trie *root)
{
    q[++tail]=root;
    while(head!=tail)
    {
        trie *p=q[++head];
        for(int i=0;i<26;i++)
        if(p->nxt[i]!=NULL)
        {
            if(p==root)p->nxt[i]->fail=p;
            else       p->nxt[i]->fail=p->fail->nxt[i];
            q[++tail]=p->nxt[i];
        }
        else
        {
            if(p==root)p->nxt[i]=p;
            else       p->nxt[i]=p->fail->nxt[i];
        }
    }
}
int ask(trie *root)
{
    trie *p=root;
    int i=0,index,cnt=0;
    while(str[i])
    {
        index=str[i]-a;
        p=p->nxt[index];
        trie *temp=p;
        while(temp!=root && temp->count!=-1)
            cnt+=temp->count,temp->count=-1,temp=temp->fail;
        i++;
    }
    return cnt;
}
signed main()
{
    int n,T;
    trie *p;
    cin>>T;
    while(T--)
    {
        p=new trie();
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>keyword;
            insert(keyword,p);
        }
        cin>>str;
        build_ac(p);
        cout<<ask(p)<<endl;
    }
}
View Code

 

模板—trie图

标签:自动机   cout   end   using   namespace   include   alt   play   ace   

原文地址:https://www.cnblogs.com/Al-Ca/p/11025281.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!