题意:给出一个串S,只包含ACGT四个字符,求有多少个长度为m的串T与S的最长公共子序列长度为i (0<=i<=|S|)。
思路:假设当前T与S匹配了j位,那么可以得到dp[i][j],即是求LCS的dp数组,由于求这个数组的时候,只需要前一位的状态,枚举一下T的下一位就可以得到匹配了j+1位的状态。由于S不长,可以考虑状态压缩来保存这个dp数组,每个状态1表示在这一位+1,预处理一下每个状态遇到每个字符后会转移到的下一个状态,就可以了,剩下只需要m次dp就行了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int mod=1000000000+7;
const char DNA[4]={'A','C','G','T'};
int dp[2][1<<15],n,m;
int next_state[1<<15][4];
int ans[20],cnt[1<<15];
char s[20];
void Init()
{
memset(cnt,0,sizeof(cnt));
for(int i=0;i<(1<<15);++i)
{
int x=i;
while(x)
{
cnt[i]++;
x=x&(x-1);
}
}
}
void solve()
{
memset(dp,0,sizeof(dp));
int cur[20],next[20];
for(int i=0;i<(1<<n);++i)
{
cur[0]=0;
for(int j=1;j<=n;++j)
cur[j]=cur[j-1]+((i>>(j-1))&1);
for(int c=0;c<4;++c)
{
next[0]=0;
for(int j=n;j>=1;--j)
if(s[j-1]==DNA[c]) next[j]=max(cur[j],cur[j-1]+1);
else next[j]=cur[j];
for(int j=1;j<=n;++j)
next[j]=max(next[j],next[j-1]);
int tmp=0;
for(int j=1;j<=n;++j)
if(next[j]>next[j-1])
tmp|=(1<<(j-1));
next_state[i][c]=tmp;
}
}
dp[0][0]=1;
for(int i=1;i<=m;++i)
{
int p=i&1;
memset(dp[p],0,sizeof(dp[p]));
for(int j=0;j<(1<<n);++j)
{
if(dp[p^1][j]==0) continue;
for(int c=0;c<4;++c)
{
dp[p][next_state[j][c]]+=dp[p^1][j];
if(dp[p][next_state[j][c]]>=mod) dp[p][next_state[j][c]]-=mod;
}
}
}
memset(ans,0,sizeof(ans));
for(int i=0;i<(1<<n);++i)
{
ans[cnt[i]]+=dp[m&1][i];
if(ans[cnt[i]]>=mod) ans[cnt[i]]-=mod;
}
}
int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
Init();
int t;
scanf("%d",&t);
while(t--)
{
scanf("%s",s);
scanf("%d",&m);
n=strlen(s);
solve();
for(int i=0;i<=n;++i)
printf("%d\n",ans[i]);
}
return 0;
}