zoukankan      html  css  js  c++  java
  • ac自动机模版(hdu2222)

    #include <iostream>
    #include <cstring>
    #include <queue>
    #include <cstdio>
    #include <algorithm>
    
    using namespace std;
    
    const int K=26;//一共有多少种字符
    const int MAXC=500010;//模版串包含的字符数量
    const int MAXN=1000010;//匹配串的最大长度
    
    struct Trie
    {
        int next[MAXC][K],fail[MAXC],end[MAXC];
        int root,L;
        int newnode()
        {
            for(int i=0;i<K;i++)
                next[L][i]=-1;
            end[L++]=0;
            return L-1;
        }
        void init()
        {
            L=0;
            root=newnode();
        }
        void insert(char buf[])
        {
            int len=strlen(buf);
            int now=root;
            for(int i=0;i<len;i++)
            {
                if(next[now][buf[i]-'a']==-1)
                    next[now][buf[i]-'a']=newnode();
                now=next[now][buf[i]-'a'];
            }
            end[now]++;
        }
        void build()
        {
            queue<int>Q;
            fail[root]=root;
            for(int i=0;i<K;i++)
                if(next[root][i]==-1)
                    next[root][i]=root;
                else
                {
                    fail[next[root][i]]=root;
                    Q.push(next[root][i]);
                }
            while(!Q.empty())
            {
                int now=Q.front();
                Q.pop();
                for(int i=0;i<K;i++)
                {
                    if(next[now][i]==-1)
                        next[now][i]=next[fail[now]][i];
                    else
                    {
                        fail[next[now][i]]=next[fail[now]][i];
                        Q.push(next[now][i]);
                    }
                }
            }
        }
        int query(char buf[])
        {
            int len=strlen(buf);
            int now=root;
            int res=0;
            for(int i=0;i<len;i++)
            {
                now=next[now][buf[i]-'a'];
                int temp=now;
                while(temp!=root)
                {
                    res+=end[temp];
                    end[temp]=0;
                    temp=fail[temp];
                }
            }
            return res;
        }
    };
    
    char buf[MAXN];
    Trie ac;
    
    int main()
    {
        int T;
        int n;
        scanf("%d",&T);
        while(T--)
        {
            scanf("%d",&n);
            ac.init();
            for(int i=0;i<n;i++)
            {
                scanf("%s",buf);
                ac.insert(buf);
            }
            ac.build();
            scanf("%s",&buf);
            printf("%d
    ",ac.query(buf));
        }
        return 0;
    }
  • 相关阅读:
    Android开发学习一:环境搭建
    Python Challenge
    Python Challenge
    Python Challenge
    Centos中的mysql安装以及配置
    linux查询日志内容命令
    搭建elasticsearch过程中遇到的问题
    启动kafka后闪退
    zookeeper环境搭建
    nginx安装
  • 原文地址:https://www.cnblogs.com/wsruning/p/5734090.html
Copyright © 2011-2022 走看看