【板子】AC自动机

//lg p3808
//Copyright yeyou26
#include<bits/stdc++.h>
using namespace std;

string mainstring;
string ss[(int)1e6+6];
int cnt[(int)1e6+6];
int n;
int idx;
int 

int t[(int)1e6+6][30],nxt[(int)1e6+5],idx,n;
string mainstring,ss[(int)1e6+6];


void init();
void Addtrie(string s);
void Trie();
queue<int> q;
void Aho_corasick();
int Query(string mainstring);

int main()
{
    init();
    Trie();
    Aho_corasick();
    cout<<Query(mainstring);
    return 0;
}

//Function Implementation

int Query(string s)
{
    int ans=0;
    int p=0;
    for(int i=0;s[i];i++)
    {
        int c=s[i]&31;
        p=t[p][c];
        for(int j=p;cnt[j]!=-1 && j;j=nxt[j])
        {
            ans+=cnt[j];
            cnt[j]=-1;
        }
    }
    return ans;
}

void Addtrie(string s)
{
    int p=0;
    for(int i=0;s[i];i++)
    {
        int c=s[i]&31;
        if(!t[p][c])
        {
        t[p][c]=++idx;
        }
        p=t[p][c];
    }
    cnt[p]++;
}

void Trie()
{
    for(int i=1;i<=n;i++)
    {
        Addtrie(ss[i]);
    }
}

void Aho_corasick()
{
    for(int i=1;i<=26;i++)
    {
        if(t[0][i]) q.push(t[0][i]);
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=1;i<=26;i++)
        {
            int v=t[u][i];
            if(v)
            {
                nxt[v]=t[nxt[u]][i];
                q.push(v);
            }
            else 
            {
                t[u][i]=t[nxt[u]][i];
            }
        }
    }
}

void init()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>ss[i];
    }
    cin>>mainstring;
}
posted @ 2024-01-26 20:56  yeyou26  阅读(16)  评论(0编辑  收藏  举报