「刷题笔记」AC自动机

最新推荐文章于 2022-03-10 16:39:21 发布

ぺdepths°

最新推荐文章于 2022-03-10 16:39:21 发布

阅读量165

点赞数

分类专栏：字符串----AC自动机模板刷题笔记

本文链接：https://blog.csdn.net/weixin_43464149/article/details/107832457

版权

模板同时被 3 个专栏收录

13 篇文章 0 订阅

订阅专栏

字符串----AC自动机

1 篇文章 0 订阅

订阅专栏

刷题笔记

1 篇文章 0 订阅

订阅专栏

题单

1.洛谷 P3808 AC自动机（简单版）
2.洛谷 P3796 AC自动机（加强版）
3.洛谷 P5357 AC自动机（二次加强版）

1.洛谷 P3808 AC自动机（简单版）

题目来源：https://www.luogu.com.cn/problem/P3808

题意： 给定 $n$ 个模式串 $s_i$ 和一个文本串 $t$ ，求有多少个不同的模式串在文本串里出现过。两个模式串不同当且仅当他们编号不同。

解析： AC自动机入门题，建立 $T r i e$ 树的时候统计下各个模式串的终点经过的次数，然后 $b f s$ 构造 $T r i e$ 树上各个节点的 $f a i l$ 指针，最后从 $T r i e$ 树的根开始，与文本串一位一位匹配。每走一位，就要找一遍以当前位置为结尾的所有后缀，也就是暴力跳一遍 $f a i l$ ，直到回到根为止，此处也可以剪枝一下，跳过的点标记一下，因为下次不会再走了。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#define ft first
#define sd second
#define pb push_back
#define endl '\n'
#define nul string::npos
#define sz(x) int(x.size())
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
const int maxn=1e6+7;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int t;
int n,m;
char str[maxn];
struct Trie{
    int ch[maxn][26],fail[maxn],vis[maxn],cnt;
    void add(char *s){ //构建 trie 树
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            int v=s[i]-'a';
            if(!ch[now][v])ch[now][v]=++cnt;
            now=ch[now][v];
        }
        vis[now]++; //统计
    }
    void build(){
        queue<int> q;
        for(int i=0;i<26;i++)if(ch[0][i])fail[ch[0][i]]=0,q.push(ch[0][i]);
        while(!q.empty()){
            int u=q.front();q.pop();
            for(int i=0;i<26;i++){
            	if(ch[u][i])fail[ch[u][i]]=ch[fail[u]][i],q.push(ch[u][i]);
            	else ch[u][i]=ch[fail[u]][i];
            }
        }
    }
    int ask(char *s,int ans=0){
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            now=ch[now][s[i]-'a'];
            for(int j=now;j&&~vis[j];j=fail[j])ans+=vis[j],vis[j]=-1;//剪枝一下
        }
        return ans;
    }
}AC;
int main(){
    cin>>n;
    while(n--){
        scanf("%s",str);
        AC.add(str);
    }
    AC.build();
    cin>>str;
    cout<<AC.ask(str)<<'\n';
	return 0;
}

2.洛谷 P3796 AC自动机（加强版）

题目来源： https://www.luogu.com.cn/problem/P3796

题意： 有 $n$ 个由小写字母组成的模式串以及一个文本串 $t$ 。每个模式串可能会在文本串中出现多次。你需要找出哪些模式串在文本串 $t$ 中出现的次数最多。

解析： 这道题数据不是很强，建出AC自动机然后每匹配到一个节点就暴力跳 $f a i l$ ，也是可以过的，考虑本题与上一题的区别在于本题的模式串每出现一次就要统计一次，所以每个节点必须跳 $f a i l$ 一直到根，无法剪枝。考虑一个字符串 $s$ 的后缀子串个数显然是 $O (∣ s ∣)$ 的，匹配文本串的复杂度是 $O (∣ t ∣)$ 的，于是总复杂度 $O (∣ s ∣ ∣ t ∣)$ 。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#define ft first
#define sd second
#define pb push_back
#define endl '\n'
#define nul string::npos
#define sz(x) int(x.size())
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
const int maxn=1e5+7;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int n,m;
char t[maxn*10];
char str[maxn][75];
int ans[maxn];

struct ACTrie{
    int c[maxn][26],fail[maxn],val[maxn],cnt;
    void init(){
        memset(c,0,sizeof(c));
        memset(val,0,sizeof(val));
        cnt=0;
    }
    void add(char *s,int id){
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            int v=s[i]-'a';
            if(!c[now][v])c[now][v]=++cnt;
            now=c[now][v];
        }
        val[id]=now;
    }
    void build(){
        queue<int> q;
        for(int i=0;i<26;i++)if(c[0][i])fail[c[0][i]]=0,q.push(c[0][i]);
        while(!q.empty()){
            int u=q.front();q.pop();
            for(int i=0;i<26;i++){
                if(c[u][i]){
                    q.push(c[u][i]);
                    fail[c[u][i]]=c[fail[u]][i];
                }
                else c[u][i]=c[fail[u]][i];
            }
        }
    }
    void query(char *s){
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            now=c[now][s[i]-'a'];
            for(int j=now;j;j=fail[j])ans[j]++;
        }
    }
}AC;
int main(){
    while(cin>>n){
        if(!n)break;
        AC.init();
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=n;i++){
            scanf("%s",str[i]);
            AC.add(str[i],i);
        }
        AC.build();
        scanf("%s",t);
        AC.query(t);
        int ma=-1;
        for(int i=1;i<=n;i++){
            ma=max(ma,ans[AC.val[i]]);
        }
        printf("%d\n",ma);
        for(int i=1;i<=n;i++){
            if(ans[AC.val[i]]==ma)printf("%s\n",str[i]);
        }
    }
	return 0;
}

3.洛谷 P5357 AC自动机（二次加强版）

题目来源：https://www.luogu.com.cn/problem/P5357

题意： 给你一个文本串 $t$ ( $t|≤2×10^6$ ) 和 $n$ 个模式串 $s_{1..n}$ ( $s|≤2×10^5$ )，请你分别求出每个模式串 $s_i$ 在 $t$ 中出现的次数。

解析： 由于模式串和文本串的长度都较大，还沿用上一题的暴力跳 $f a i l$ 复杂度为 $O (∣ s ∣ ∣ t ∣)$ 的做法肯定超时。我们发现，在暴跳的过程中， $f a i l$ 的形态是一棵树，试想从一个点出发暴跳 $f a i l$ 的过程，最终肯定会到达 $r o o t$ ，而在这条路径上某些点会有 $1$ 的贡献，所以我们不妨建一棵 $f a i l$ 树，也就是让每个点和它的 $f a i l$ 指针指向的结点连一条边，把 $r o o t$ 作为根，那么每个点暴跳 $f a i l$ 的路径就在这棵 $f a i l$ 树上展现的淋漓尽致。又因为我们已经标记了会有贡献的那些点(在建 $T r i e$ 树的时候)，所以我们可以在遍历文本串的时候统一把每个点到根的路径贡献都加一，也就是把路径上的点的贡献都统一为 $1$ ，就算有些点本来的贡献不是 $1$ ，但是并不影响我们统计答案。这样的话，我们就可以利用树上差分(单链修改，单点查询)来解决这道题了。差分之后， $d f s$ 一下，子树大小即为该点贡献。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#define ft first
#define sd second
#define pb push_back
#define endl '\n'
#define nul string::npos
#define sz(x) int(x.size())
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
const int maxn=2e5+7;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int n,m;
char t[maxn*10];
char str[maxn];
int ans[maxn];

struct Trie{
    int c[maxn][26],fail[maxn],val[maxn],cnt;
    vector<int> g[maxn];
    void add(char *s,int id){
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            int v=s[i]-'a';
            if(!c[now][v])c[now][v]=++cnt;
            now=c[now][v];
        }
        val[id]=now;
    }
    void build(){
        queue<int> q;
        for(int i=0;i<26;i++)if(c[0][i]){
        	fail[c[0][i]]=0,q.push(c[0][i]);
        	g[0].pb(c[0][i]),g[c[0][i]].pb(0);//结点和其fail指向的结点连一条边
        }
        while(!q.empty()){
            int u=q.front();q.pop();
            for(int i=0;i<26;i++){
                if(c[u][i]){
                    q.push(c[u][i]);
                    fail[c[u][i]]=c[fail[u]][i];
                    g[c[u][i]].pb(fail[c[u][i]]);//结点和其fail指向的结点连一条边
                    g[fail[c[u][i]]].pb(c[u][i]);
                }
                else c[u][i]=c[fail[u]][i];
            }
        }
    }
    void dfs(int u,int fa){ //简单树形dp统计子树大小
        for(int i=0;i<sz(g[u]);i++){
            int v=g[u][i];
            if(v==fa)continue;
            dfs(v,u);
            ans[u]+=ans[v];
        }
    }
    void ask(char *s){
        int len=strlen(s),now=0;
        for(int i=0;i<len;i++){
            now=c[now][s[i]-'a'];
            ans[now]++;//由于根的贡献不用考虑且lca(root,now)=root，所以只让now端贡献加1即可
        }
        dfs(0,-1);
    }
}AC;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%s",str);
        AC.add(str,i);
    }
    AC.build();
    scanf("%s",t);
    AC.ask(t);
    for(int i=1;i<=n;i++){
        printf("%d\n",ans[AC.val[i]]);
    }
	return 0;
}

未完待续…

ぺdepths°

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
「刷题笔记」AC自动机

单调栈一.原理二.习题练习1.洛谷 P3808 AC自动机（简单版）2.洛谷 P3796 AC自动机（加强版）3.洛谷 P5357 AC自动机（二次加强版）4.5.一.原理二.习题练习1.洛谷 P3808 AC自动机（简单版）题目来源：https://www.luogu.com.cn/problem/P3808题意：解析：#include <iostream>#include <cstdio>#include <cstring>#include &lt
复制链接

扫一扫