題目鏈接:http://zerojudge.tw/ShowProblem?problemid=b179
題目大意:一個基因可以分裂出四個子基因,這四個基因末尾增加a,b,c,或d,但本身的第一個鹼基會失去,當基因長度爲0時會死去,同時當基因含有制病片段時就會生病,問p天后有好多死去,好多生病。
題目思路:ac自動機dp,dp[i][j][k]表示第i天在結點k有多少長度爲j的基因。對於基因變短,可以由fail指針轉移。要分兩種
情況討論,長度小於結點深度和不小於結點深度。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
#define mod 10007
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
struct node
{
int cnt,fail,dep;
int next[4];
void init()
{
cnt=fail=dep=0;
memset(next,0,sizeof(next));
}
}tri[1600];
char s[1000],str[1000];
int dp[330][110][1600],q[1600];
int cnt;
void insert(char *s)
{
int i,p,x;
p=0;
for(i=0;s[i];i++)
{
x=s[i]-'a';
if(!tri[p].next[x])
{
tri[++cnt].init();
tri[cnt].dep=tri[p].dep+1;
tri[p].next[x]=cnt;
}
p=tri[p].next[x];
}
tri[p].cnt++;
}
void bfs()
{
int i,p,head,tail,suf;
p=head=tail=0;
for(i=0;i<4;i++)
{
if(tri[0].next[i])
{
q[tail++]=tri[0].next[i];
tri[q[tail-1]].fail=0;
}
}
while(head<tail)
{
p=q[head++],suf=tri[p].fail;
tri[p].cnt+=tri[suf].cnt;
for(i=0;i<4;i++)
{
if(tri[p].next[i])
{
q[tail++]=tri[p].next[i];
tri[q[tail-1]].fail=tri[suf].next[i];
}
else
tri[p].next[i]=tri[suf].next[i];
}
}
}
void solve(char *s,int n)
{
int i,j,k,g,p,x,pos;
int len=strlen(s);
p=0;
for(i=0;i<=n;i++)
for(j=0;j<=100&&j<=n+len;j++)
for(k=0;k<=cnt;k++)
dp[i][j][k]=0;
for(i=0;s[i];i++)
{
x=s[i]-'a';
p=tri[p].next[x];
if(tri[p].cnt)
{
printf("0 1\n");
return;
}
}
dp[0][len][p]=1;
for(i=0;i<n;i++)
for(j=1;j<=100&&j<=n+len;j++)
for(k=0;k<=cnt;k++)
{
if(dp[i][j][k]==0||tri[k].cnt) continue;
for(g=0;g<4;g++)
{
pos=tri[k].next[g];
dp[i+1][j+1][pos]+=dp[i][j][k];
dp[i+1][j+1][pos]%=mod;
}
if(j-1<tri[k].dep)
{
pos=tri[k].fail;
dp[i+1][j-1][pos]+=dp[i][j][k];
dp[i+1][j-1][pos]%=mod;
}
else
{
dp[i+1][j-1][k]+=dp[i][j][k];
dp[i+1][j-1][k]%=mod;
}
}
int ans1=0,ans2=0;
for(i=1;i<=n;i++)
ans1=(ans1+dp[i][0][0])%mod;
for(i=1;i<=n;i++)
for(j=0;j<=100&&j<=n+len;j++)
for(k=0;k<=cnt;k++)
{
// printf("i %d j %d k %d dp %d\n",i,j,k,dp[i][j][k]);
if(tri[k].cnt)
ans2=(ans2+dp[i][j][k])%mod;
}
printf("%d %d\n",ans1,ans2);
}
int main()
{
int p,n,i,j;
while(scanf("%s",s)!=EOF)
{
scanf("%d%d",&p,&n);
cnt=0;
tri[0].init();
while(n--)
{
scanf("%s",str);
insert(str);
}
bfs();
solve(s,p);
}
}