【BZOJ3864】Hero meet devil（dp）

题目大意

对每一个i(1<=i<=n)，求长度为m，与给定字符串S的最长公共子序列的长度为i的字符串有多少个？

题解

DP新套路
刚开始想的时候，怎么定义状态都会造成重复等各种问题，于是搜题解。。。

考虑求LCS时的dp：定义$lcs[i][j]$表示A串的前i位与B串的前j位的LCS长度
$$lcs[i][j]=max\{\begin{array}{l}lcs[i-1][j-1]+1(A[i]==B[i])\\ lcs[i][j-1]\\ lcs[i-1][j]\end{array}$$
可以发现，当i相同时，lcs[i][j]与lcs[i][j+1]最多相差1
把每一位之间的差状压，表示为一个状态进行计数，就可以了。

定义$dp[i][s]$表示长度为i的字符串，与制定串S的lcs状态为s的方案数
$trans(s,k)$表示在lcs状态为s的情况下，添加字符k转移到的状态
$$dp[i][trans(s,k)]+=dp[i-1][s]$$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=1005,MAXL=20,MAXS=(1<<15)+10,MOD=1000000007;
const char ID[]="ACGT";

int T,n,m,dp[2][MAXS],tr[MAXS][4],ans[MAXN];
char str[MAXL];
int f[MAXL],g[MAXL];

int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s%d",str+1,&m);
		n=strlen(str+1);
		for(int s=0;s<(1<<n);s++)
		{
			memset(f,0,sizeof f);
			memset(g,0,sizeof g);
			for(int j=1;j<=n;j++)
				f[j]=f[j-1]+((s>>(j-1))&1);
			for(int k=0;k<4;k++)
			{
				for(int j=1;j<=n;j++)
				{
					g[j]=max(g[j-1],f[j]);
					if(ID[k]==str[j])
						g[j]=max(g[j],f[j-1]+1);
				}
				tr[s][k]=0;
				for(int j=1;j<=n;j++)
					if(g[j]-g[j-1])
						tr[s][k]|=1<<(j-1);
			}
		}
		memset(dp,0,sizeof dp);
		dp[0][0]=1;
		for(int i=1;i<=m;i++)
		{
			int p=i&1;
			memset(dp[p],0,sizeof dp[p]);
			for(int s=0;s<(1<<n);s++)
				for(int k=0;k<4;k++)
					(dp[p][tr[s][k]]+=dp[p^1][s])%=MOD;
		}
		memset(ans,0,sizeof ans);
		for(int s=0;s<(1<<n);s++)
			(ans[__builtin_popcount(s)]+=dp[m&1][s])%=MOD;
		for(int i=0;i<=n;i++)
			printf("%d\n",ans[i]);
	}
	return 0;
}