POJ 2778 AC自动机+矩阵乘法

原创 2012年03月28日 22:12:19

这题要写的东西很多,以至于我不知道从哪里开始写了= =。

总之要正确理解AC自动机中的状态转移。时间比较晚了,早点回去... 

我要注意的地方就是正确理解fail指针的作用。在失配时,fail指针如果走向的是一个病毒节点那么改节点的转移也是病毒节点。不能转移!

另外要注意的就是mod十分的耗时间啊!两次TLE都是出于此....

总之是一道好题

#include<iostream>
#include<cstdio>
#include<string.h>
#define kind 4
#define MAX 111
using namespace std;

struct node
{
 	   int fail,flag;
 	   int next[4];
 	   void init(){
	   		fail=flag=0;
	   		memset(next,0,sizeof(next));
	   }
}Tire[MAX];

__int64 matrix[MAX][MAX];
int root,allocp;
__int64 tot[MAX][MAX];
  		   
int index(char c)
{
 	if( c=='A' )return 0;
 	if( c=='C' )return 1;
 	if( c=='G' )return 2;
 	if( c=='T' )return 3;
}

void init()
{
 	 root=allocp=0;
 	 Tire[root].init();
 	 memset( matrix,0,sizeof(matrix) );
	 memset(tot,0,sizeof(tot));
}

void Insert( char *s ){
 	 int i=0,k;
 	 int p=root;
 	 while( s[i] ){
	  		k=index(s[i++]);
	  		if( !Tire[p].next[k] ){
			 	Tire[++allocp].init();
			 	Tire[p].next[k]=allocp;
			}else if( Tire[p].flag )
				  return ;
			p=Tire[p].next[k];
	 }
	 Tire[p].flag++;
}

void build_ac_automation()
{
 	 int queue[111];
 	 int head,foot;
 	 head=foot=0;
 	 queue[foot++]=root;
 	 while( head<foot ){
	  		int cur=queue[head++];
	  		for( int i=0;i<kind;i++ )
			{
 	 		 	 int son=Tire[cur].next[i];
 	 			 if( son ){//if have next
				  	 int p=Tire[cur].fail;
				  	 if( cur==root ) Tire[son].fail=root;
				  	 else Tire[son].fail=Tire[p].next[i];
				  	 if( Tire[Tire[son].fail].flag )
				  	 	 Tire[son].flag++;
				  	 queue[foot++]=son;
			   	 }
			   	 else{
				  	 int p=Tire[cur].fail;
				  	 if( cur==root ) Tire[cur].next[i]=root;
				  	 else Tire[cur].next[i]=Tire[p].next[i];
			   	 }
 	 	 	}
	 }
}

void set_matrix()
{
 	 int son;
 	 for( int p=0;p<=allocp;p++ ){
	  	  if( Tire[p].flag )
	  	  	  continue;
	  	  for( int i=0;i<kind;i++ ){
		   	   son=Tire[p].next[i];
		   	   if( Tire[son].flag==0 )
		   	   	   matrix[p][son]++;
  	   	  }
	 }/*
	 for( int i=0;i<=allocp;i++ ){
	  	  for( int j=0;j<=allocp;j++ )
	 	  	   printf( "%d",matrix[i][j] );
	 	  printf( "\n" );
     }*/
}

__int64 res[MAX][MAX],temp[MAX][MAX];
 	 
void matrixMult( __int64 a[][MAX],__int64 b[][MAX] )
{
 	 __int64 kk[MAX][MAX];
 	 memset( kk,0,sizeof(kk) );
 	 for( int i=0;i<=allocp;i++ )
 	 for( int j=0;j<=allocp;j++ )
 	 for( int k=0;k<=allocp;k++ )
 	 	  kk[i][j]+=a[i][k]*b[k][j];
 	 	  
 	 for( int i=0;i<=allocp;i++ )
 	 for( int j=0;j<=allocp;j++ )
 	 	  a[i][j]=kk[i][j]%100000;
}


__int64 matrix_Power(int n)
{
 	 memset( res,0,sizeof(res) );
 	 memset( temp,0,sizeof(temp) );
 	 int ret=0;
 	 for( int i=0;i<=allocp;i++ )
 	 {
 	 	  res[i][i]=1;
 	 	  for( int j=0;j<=allocp;j++ )
 	 	  	   temp[i][j]=matrix[i][j];
	 }
 	 for( int p=0;p<32;p++ ){
	 	  if( n&(1<<p) )
	 	  	  matrixMult( res,temp );
	 	  matrixMult( temp,temp );
	 }
	 for( int i=0;i<=allocp;i++ )
	 	  ret+=res[0][i];
	 return ret%100000;
}


int main(){
	//freopen( "my.out","w",stdout );
 	int m,n;
 	while( scanf("%d%d",&m,&n)!=EOF ){
		   init();
	 	   char str[22];
	 	   while( m-- ){
		   		scanf("%s",&str);
				Insert(str);
  		   }
  		   build_ac_automation();
  		   set_matrix();
  		   __int64 sum=matrix_Power(n);
  		   printf( "%d\n",sum%100000 );
  	}
  	return 0;
}


版权声明:本文为博主原创文章,转载请写明出处。

相关文章推荐

poj2778 DNA Sequence AC自动机 dp 矩阵乘法

poj2778 DNA Sequence 题意:求仅有A G C T不含有k(k
  • yxr0105
  • yxr0105
  • 2016年02月03日 01:02
  • 176

Poj 2778 [AC自动机,矩阵乘法]

-----我的AC自动机----- •题意:有m种DNA序列是有疾病的,问有多少种长度为n的DNA序列不包含任何一种有疾病的DNA序列。(仅含A,T,C,G四个字符) •样例m=4,n=3...

poj2778 DNA Sequence AC自动机+矩阵乘法

DNA Sequence Time Limit: 1000MS Memory Limit: 65536K Total Submissions: 15900 Accepted...

poj 2778 DNA Sequence //AC自动机+矩阵乘法

DNA SequenceTime Limit: 1000MS Memory Limit: 65536KTotal Submissions: 5080 Accepted: 1766Description...
  • hqd_acm
  • hqd_acm
  • 2011年04月22日 21:59
  • 1219

hdu 2243 poj 2778 AC自动机 + 经典矩阵乘法

两个题差不多,只不过hdu上的恶心一点,对2^64取模,其实就相当于不用取模,所有的数都用unsigned __int64 即可,注意,无符号类型6-9就不是-3了哦 经典矩阵乘法是指求A+A^2+...

【POJ2778】AC自动机,DP,矩阵乘法

题意:给出n个字串表示“缺陷基因”,然后让求长度为m的基因(4^m个)中有多少个不带病。 题解:首先建立AC自动机,然后从每个节点开始选“ATGC”有四种往外转移的途径。 如:ACG,C这...
  • Vmurder
  • Vmurder
  • 2014年09月17日 08:13
  • 1068

poj 2778 DNA Sequence 【ac自动机 + dp + 矩阵快速幂】

这题基本上是纯模版,开始wa了,然后和模版对照了一下,居然是建fail指针的时候搞错了。。。少写了一个q.push(to) #include #include #include #include...
  • zz_1215
  • zz_1215
  • 2012年05月17日 11:01
  • 688

poj2778 AC自动机+矩阵快速幂

推荐博客:这里

POJ2778 DNA Sequence AC自动机+矩阵快速幂

题目描述:给出m(m        这道题目深层次地利用了AC自动机,但是并没有用到AC自动机中的find(query)函数,而是用到了BFS序建立失败指针走向,以及为字典树中的空节点加上...
  • jijijix
  • jijijix
  • 2017年02月17日 15:32
  • 292

【POJ】2778 DNA Sequence AC自动机+矩阵

DNA Sequence Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 11143   ...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:POJ 2778 AC自动机+矩阵乘法
举报原因:
原因补充:

(最多只允许输入30个字)