POJ2778 AC自动机 + 矩阵

最新推荐文章于 2022-02-25 19:38:17 发布

x269390114

最新推荐文章于 2022-02-25 19:38:17 发布

阅读量382

点赞数 1

分类专栏： AC自动机文章标签： null insert build struct c

本文链接：https://blog.csdn.net/x269390114/article/details/8040330

版权

AC自动机专栏收录该内容

0 篇文章 0 订阅

订阅专栏

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
#define mod 100000
struct trie{
	trie *next[4];
	trie *fail;
	int tag,isdanger;
	trie(){
		fail=NULL;
		memset(next,NULL,sizeof(next));
		tag=isdanger=0;
	}
}*q[111];
int res=0;
struct MT{
	__int64 mat[101][101];
};

int dis(char x){
	if(x=='A') return 0;
	else if(x=='C') return 1;
	else if(x=='T') return 2;
	else return 3;
}

void insert(trie *root,char *str){
	int index,i=0;
	trie *p=root;
	while(str[i]){
		index=dis(str[i]);
		if(p->next[index]==NULL) p->next[index]=new trie(),p->next[index]->tag=res++;
		p=p->next[index];
		i++;
	}
	p->isdanger=1;
}

void build_ac(trie *root){
	int i,head,tail;
	head=tail=0;
	q[head++]=root;
	while(head!=tail){
		trie *temp=q[tail++];
		trie *p=NULL;
		temp->isdanger=temp->isdanger||temp->fail->isdanger;
		for(i=0;i<4;i++){
			if(temp->next[i]==NULL){
				if(temp==root) temp->next[i]=root;
				else temp->next[i]=temp->fail->next[i];//有问题 
			}
			else{	
				if(temp==root) temp->next[i]->fail=root;
				else{
					p=temp->fail;
					while(p!=NULL){
						if(p->next[i]!=NULL){
							temp->next[i]->fail=p->next[i];
							break;
						}
						p=p->fail;
					}
					if(p==NULL) temp->next[i]->fail=root;
				}
				q[head++]=temp->next[i];
			}
		}
	}
}

void build_mat(MT &pr){
	int i,j;
	memset(pr.mat,0,sizeof(pr.mat));
	for(i=0;i<res;i++){
		if(q[i]->isdanger==0)
		for(j=0;j<4;j++){
			if(q[i]->next[j]->isdanger==0)
				pr.mat[q[i]->tag][q[i]->next[j]->tag]++;	
		}
	}
	/*for(i=0;i<res;i++){
		for(j=0;j<res;j++)
			cout<<mat[i][j]<<" ";
		cout<<endl;
	}*/
}

MT mat_mult(MT a,MT b){
	int i,j,k;
	MT ans;
	memset(ans.mat,0,sizeof(ans.mat));
	for(i=0;i<res;i++)
		for(j=0;j<res;j++)
			if(a.mat[i][j]){
				for(k=0;k<res;k++){
					ans.mat[i][k]+=(a.mat[i][j]%mod)*(b.mat[j][k]%mod);
					ans.mat[i][k]%=mod;
				}
			}
	return ans;
}

MT mat_power(MT a,int b){
	if(b==1) return a;
	MT temp=mat_power(a,b/2);
	temp=mat_mult(temp,temp);
	if(b&1) return mat_mult(a,temp);
	else return temp;
}

int main(){
	int n,m;
	char ch[20];
	scanf("%d%d",&n,&m);
	trie *root=new trie();
	root->fail=root;
	root->tag=res++;
	while(n--){
		scanf("%s",ch);
		insert(root,ch);
	}
	build_ac(root);
	MT pr;
	build_mat(pr);
	MT mt=mat_power(pr,m);
	int ans=0,i;
	for(i=0;i<res;i++){
		ans+=mt.mat[0][i];
		ans%=mod;
	}
	printf("%d\n",ans);
}

/*
2 3
ACG
C
*/