POJ 2778 AC自己主动机+矩阵幂 不错的题

http://poj.org/problem?id=2778

有空再又一次做下,对状态图的理解非常重要

题解:

http://blog.csdn.net/morgan_xww/article/details/7834801

另外做了矩阵幂的模板:

//ac.sz是矩阵的大小
void mulmtr(long long x[MAXNODE][MAXNODE],long long y[MAXNODE][MAXNODE])//y=x*y
{
ll tmp[MAXNODE][MAXNODE];
for(int i=0;i<ac.sz;i++)
{
for(int j=0;j<ac.sz;j++)
{
tmp[i][j]=0;
for(int k=0;k<ac.sz;k++)
tmp[i][j] +=x[i][k]*y[k][j];
tmp[i][j] %=MOD;
}
}
for(int i=0;i<ac.sz;i++)
for(int j=0;j<ac.sz;j++)
y[i][j]=tmp[i][j];
}
void Mtrmi(ll mtr[MAXNODE][MAXNODE],int n)
{
for(int i=0;i<ac.sz;i++)
{
for(int j=0;j<ac.sz;j++)
{
if(i == j)ans[i][j]=1;//E矩阵
else ans[i][j]=0;
}
}
while(n)
{
if(n&1)
{
mulmtr(mtr,ans);
}
mulmtr(mtr,mtr);
n/=2;
}
}

代码:

#include <cstdio>
#include <cstring>
#include <string>
#include <map>
#include <queue>
#include <iostream>
using namespace std; #define ll long long const int MAXNODE = 15*15;
const int SSIZE = 2000000000+100;
const int MOD = 100000;
const int SIGMA_SIZE = 4;
const int SIZE = 20; ll mtr[MAXNODE][MAXNODE];
ll ans[MAXNODE][MAXNODE];
int danger[MAXNODE]; struct AC
{
int f[MAXNODE];
int val[MAXNODE];
int last[MAXNODE];
int cnt[MAXNODE];
int ch[MAXNODE][SIGMA_SIZE];
int sz; void init()
{
memset(ch[0],0,sizeof(ch[0]));
memset(cnt,0,sizeof(cnt));
f[0]=0;///////////
sz=1;
} inline int idx(char x)
{
if(x == 'A')return 0;
if(x == 'T')return 1;
if(x == 'C')return 2;
if(x == 'G')return 3;
} void insert(char *s, int v)
{
int n=strlen(s),u=0;
for(int i=0;i<n;i++)
{
int id= idx(s[i]);
if(!ch[u][id])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][id]=sz++;
}
u=ch[u][id];
}
val[u]=v;
danger[u]=1;////////
} void getfail()
{
queue<int>q;
f[0]=0;
for(int c=0;c<SIGMA_SIZE;c++)
{
int u=ch[0][c];
if(u)
{
q.push(u);
f[u]=0;
last[u]=0;
}
}
while(!q.empty())
{
int r=q.front();q.pop();
for(int c=0;c<SIGMA_SIZE;c++)
{
int u=ch[r][c];
//if(!u)continue;////////
if(!u)
{
ch[r][c]=ch[f[r]][c];//////
continue;
}
q.push(u);
int v=f[r];
while(v &&!ch[v][c])v=f[v];
f[u]=ch[v][c];
//last[u]=val[f[u]]?f[u]:last[f[u]];
danger[u] |= danger[f[u]];
}
}
}
};
void init()
{
memset(mtr,0,sizeof(mtr));
memset(danger,0,sizeof(danger));
} AC ac; char str[SIZE]; void mulmtr(long long x[MAXNODE][MAXNODE],long long y[MAXNODE][MAXNODE])//y=x*y
{
ll tmp[MAXNODE][MAXNODE];
for(int i=0;i<ac.sz;i++)
{
for(int j=0;j<ac.sz;j++)
{
tmp[i][j]=0;
for(int k=0;k<ac.sz;k++)
tmp[i][j] +=x[i][k]*y[k][j];
tmp[i][j] %=MOD;
}
}
for(int i=0;i<ac.sz;i++)
for(int j=0;j<ac.sz;j++)
y[i][j]=tmp[i][j];
}
void Mtrmi(ll mtr[MAXNODE][MAXNODE],int n)
{
for(int i=0;i<ac.sz;i++)
{
for(int j=0;j<ac.sz;j++)
{
if(i == j)ans[i][j]=1;//E矩阵
else ans[i][j]=0;
}
}
while(n)
{
if(n&1)
{
mulmtr(mtr,ans);
}
mulmtr(mtr,mtr);
n/=2;
}
} int main()
{
//freopen("poj2788.txt","r",stdin);
int n,m;
while(~scanf("%d%d",&m,&n))
{
init();
ac.init();
for(int i=1;i<=m;i++)
{
scanf("%s",str);
ac.insert(str,i);
}
ac.getfail();
for(int i=0;i<ac.sz;i++)
if(!danger[i])
for(int j=0;j<4;j++)
if(!danger[ac.ch[i][j]])
{
mtr[i][ac.ch[i][j]]++;
} Mtrmi(mtr,n);
/////////////////////////////////
/* for(int i=0;i<ac.sz;i++)
{
for(int j=0;j<ac.sz;j++)
printf("%lld|%lld ",mtr[i][j],ans[i][j]);
putchar('\n');
}*/
///////////////////////
for(int i=1;i<ac.sz;i++)
ans[0][0]+=ans[0][i]%MOD;
printf("%I64d\n",ans[0][0]%MOD);
}
return 0;
}
上一篇:仿iReader切换皮肤进度条


下一篇:【转载】看*如何用25台服务器撑起5.6亿的月PV