【BZOJ5336】[TJOI2018]party（动态规划）

【BZOJ5336】[TJOI2018]party（动态规划）

题面

BZOJ
洛谷

题解

这题好神仙啊。。。
考虑普通的\(LCS\)的\(dp\)，\(f[i][j]=\max\{f[i-1][j],f[i][j-1],f[i-1][j-1]+(A_i==B_j)\}\)
发现对于固定的\(i\)而言，随着\(j\)的增长，相邻的两个数之间的差不超过\(1\)，因此直接考虑一个\(2^k\)的状态记录差分的结果。
那么设\(f[i][S]\)表示当前考虑到了第\(i\)位，\(LCS\)的\(dp\)的差分数组为\(S\)的方案数。转移的时候枚举放那个字母就好了。因为还有连续字符的限制，所以再加一维就没有问题了。
这里转移因为会涉及差分数组的修改，所以可以提前把差分数组的转移给预处理出来，
然后大力\(dp\)就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MOD 1000000007
#define MAX 1010
char ch[20],QwQ[5]="NOI";
int f[2][1<<15][3],tr[1<<15][3],cnt[1<<15];
int n,k,ans[MAX];
void add(int &x,int y){x+=y;if(x>=MOD)x-=MOD;}
int dp[16],tmp[16];
int Get(int S,int c)
{
	for(int i=1;i<=k;++i)dp[i]=dp[i-1]+((S>>(i-1))&1);
	for(int i=1;i<=k;++i)
		if(QwQ[c]==ch[i])tmp[i]=dp[i-1]+1;
		else tmp[i]=max(dp[i],tmp[i-1]);
	int ret=0;
	for(int i=1;i<=k;++i)ret|=(tmp[i]-tmp[i-1])<<(i-1);
	return ret;
}
int main()
{
	scanf("%d%d",&n,&k);scanf("%s",ch+1);
	for(int i=0;i<1<<k;++i)
		for(int c=0;c<3;++c)
			tr[i][c]=Get(i,c);
	for(int i=0;i<(1<<k);++i)cnt[i]=cnt[i>>1]+(i&1);
	f[0][0][0]=1;
	for(int i=1,nw=1,pw=0;i<=n;++i,nw^=1,pw^=1)
	{
		memset(f[nw],0,sizeof(f[nw]));
		for(int j=0;j<1<<k;++j)
			for(int p=0;p<3;++p)
				for(int c=0;c<3;++c)
				{
					int np;
					if(c==0)np=1;else if(c==1)np=p==1?2:0;else if(c==2)np=p==2?3:0;
					if(np==3)continue;
					add(f[nw][tr[j][c]][np],f[pw][j][p]);
				}
	}
	for(int i=0;i<1<<k;++i)
		for(int c=0;c<3;++c)
			add(ans[cnt[i]],f[n&1][i][c]);
	for(int i=0;i<=k;++i)printf("%d\n",ans[i]);
	return 0;
}