codeforce 118D Caesar's Legions

原题地址

题意

将军有N1个步兵，N2个骑兵，将军把他们排成一列，要求不超过K1个步兵站在一起，不超过K2个骑兵站在一起，问总共的列队方案

要求所有士兵都要参与列队

题解

DP

瞎搞了一个DP

DP[i][j][k][x]：方案数

i:当前队列的长度

j:当前队列末尾连续的人数

k:当前队列中步兵人数

x:当前末尾是骑兵还是步兵

最后答案sum:DP[n1+n2][i][n1][1]+sum:DP[n1+n2][i][n1][2]

状态转移方程怪复杂的，就看代码吧

#include<bits/stdc++.h>

using namespace std;
const int h=1e8;

int dp[250][30][105][3];

int main(void)
{
	#ifdef ex1
	freopen ("in.txt","r",stdin);
	#endif
	
	int n1,n2,k1,k2;
	
	scanf("%d%d%d%d",&n1,&n2,&k1,&k2);
	
	memset(dp,0,sizeof(dp));
	dp[1][1][1][1]=1;
	dp[1][1][0][2]=1;
	
	//if (k1==1) dp[1][1][1]=0;
	//if (k2==1) dp[1][1][2]=0;
	
	for (int i=2;i<=n1+n2;++i)
	{
		for (int k=0;k<=min(i,n1);++k)
		{
			for (int p=1;p<=min(i-1,k2);++p)
		    {
			    if (k>=1) dp[i][1][k][1]=(dp[i][1][k][1]+dp[i-1][p][k-1][2])%h;
		    }
		    for (int j=2;j<=min(i,k1);++j)
		    {
			    if (k>=1) dp[i][j][k][1]=dp[i-1][j-1][k-1][1];
		    }
		    for (int p=1;p<=min(i-1,k1);++p)
		    {
			     dp[i][1][k][2]=(dp[i][1][k][2]+dp[i-1][p][k][1])%h;
		    }
		    for (int j=2;j<=min(i,k2);++j)
		    {
			    dp[i][j][k][2]=dp[i-1][j-1][k][2];
		    }
		}
	}
	
	int ans=0;
	for (int i=1;i<=min(n1+n2,k1);++i)
	{
		ans=(ans+dp[n1+n2][i][n1][1])%h;
	}
	for (int i=1;i<=min(n1+n2,k2);++i)
	{
		ans=(ans+dp[n1+n2][i][n1][2])%h;
	}
	
	printf("%d\n",ans);
	
}

注意各个状态量的边界情况