P3808 【模板】AC自动机（简单版）

题目链接

参考大佬的博客

思路：ac自动机的题主要分为3个部分
1：建立字典树（与字典树相比这里我们需要一个记录单词结尾的数组mark[ ]）
2：建立fail指针（fail[i]为与以i节点为结尾的串的后缀有最大公共长度的前缀的结尾编号）比如 fail[i]=j，那么 word2[0]到word[j]是 word1[0]到word1[i]的子串。
3.查询

#include<bits/stdc++.h>
using namespace std;

char text[1000007];//匹配串
char s[1000007];//子串

int tire[1000010][26];//字典树，i表示长度，j表示26个字母，里面存放的时节点的编号
int fail[1000010];//fail指针
int mark[1000010];//词典结束的标志
int pos = 1;//表示节点编号，从1开始，0时根节点


void Insert(char s[])//构造字典树
{
    int u = 0, len = strlen(s);
    for (int i = 0;i<len;i++)
    {
        int n = s[i] - 'a';
        if(tire[u][n]==0)
            tire[u][n] = pos++;
        u = tire[u][n];
    }
    mark[u]++;//一个单词结束位置
}

void build()//构造fail指针
{
    queue<int> q;
    for (int i = 0;i<26;i++)//寻找字典树第一层的节点，也就是每个子串的第一个字母，把他们指向根节点（0）,并入队列
    {
        if(tire[0][i]!=0)
        {
            fail[tire[0][i]] = 0;
            q.push(tire[0][i]);
        }
    }

    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        for (int i = 0;i<26;i++)
        {
            if(tire[u][i]!=0)
            {
                fail[tire[u][i]] = tire[fail[u]][i];//指向父节点的fail指针指向的节点 ,好好理解 这句代码 ,可以结合具体的列子
                q.push(tire[u][i]);
            }
            else
            {
                tire[u][i] = tire[fail[u]][i];
            }
        }
    }
}

int slove(char s[])
{
    int len = strlen(s);
    int u = 0, ans = 0;//u表示当前层数
    for (int i = 0;i<len;i++)
    {
        u = tire[u][s[i] - 'a'];
        for(int j = u; j&&~mark[j] ;j=fail[j])//应为这里是计算出现子串的数量，所以不需要重复计算~表示按位取反   ~（-1）==0
        {
            ans += mark[j];
            mark[j] = -1;
        }
    }
    return ans;
}



int main()
{
   // freopen("11.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = 0;i<n;i++)
    {
        scanf("%s", s);
        Insert(s);
    }
    build();
    
    // for (int i = 0;i<=10;i++)
    // {
    //     cout << "fali[]  " << i << "  ==" << fail[i] << endl;
    // }

    // for (int i = 0;i<=10;i++)
    // {
    //     cout << "mark[]  " << i << "  ==" << mark[i] << endl;
    // }

    scanf("%s", text);
    cout<<slove(text) << endl;

    return 0;
}