la 3942 Rember_前缀树

AC自动机实战代码解析

最新推荐文章于 2021-06-02 19:53:52 发布

转载最新推荐文章于 2021-06-02 19:53:52 发布 · 69 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/neng18/p/3676462.html

本文提供了一个AC自动机的实战代码实现，并通过一个具体的案例展示了如何使用AC自动机进行字符串匹配，包括插入字符串到自动机中及搜索过程中的计数更新等关键步骤。

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std; 
#define N 400010
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int ch[N][30];
int val[N];
int cnt;
char str[N];
int d[N];
int idx(int c){
	return c-'a';
}
void insert(char *s){
	int i,n=strlen(s);
	int c,u=0;
	for(i=0;i<n;i++){
		c=idx(s[i]);
		if(!ch[u][c]){
			memset(ch[cnt],0,sizeof(ch[cnt]));
			val[cnt]=0;
			ch[u][c]=cnt++;
		}
		u=ch[u][c];
	}
	val[u]=n;
}
void search(char *s,int i,int len){
	int j;
	int c,u=0;
	for(j=0;j<len;j++){
		c=idx(s[j]);
		if(!ch[u][c])
			break;
		u=ch[u][c];
		if(val[u])
			d[i]=(d[i]+d[i+val[u]])%20071027;
	}
}
int main(int argc, char** argv) {
	int num=1;
	int n,i,k;
	char s[N];
	while(scanf("%s",str)!=EOF){
		cnt=1;
		val[0]=0;
		memset(ch[0],0,sizeof(ch[0]));
		n=strlen(str);
		scanf("%d",&k);
		while(k--){
			scanf("%s",s);
			insert(s);
		}
		memset(d,0,sizeof(d));
		d[n]=1;
		for(i=n-1;i>=0;i--)
			search(str+i,i,n-i);
		printf("Case %d: %d\n",num++,d[0]);
	}
	return 0;
}

转载于:https://www.cnblogs.com/neng18/p/3676462.html