牛客练习赛35 B(简单dp,ORZ)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_39562952/article/details/85394855

思路：想明白了发现真的简单dp。。。我还是太菜了

代码如下：

/*
果不其然。简单dp...是我菜，不是题目难
做这题首先我没想到dp(哭晕)
看了题解半天。一脸懵逼
既然dp解那么先考虑最优解的结构特征
先想只用一维表示,dp[i]:在i这个位置上得到的最多的方案数
上面是不行的，因为dp[i]没有记录i位置上是什么元素，导致下一步无法转移(条件中说了，不能有超过连续A个元音,只用dp[i]中的信息无法知道，下一位能取什么);
既然一维不行，在动态规划的思考过程中，解决的方案应该是提高维数，从在状态中获得更多的信息
观察一维失败的原因，是不知道上一位放的是什么，所以给每个状态多两个位置存是取什么值
dp[i][0,1]:0代表这个位上是元音;1是辅音
(大佬们用这么多就够了。。。但是我想不通，所以小弟再加一维)
dp[i][0,1][j]:j代表算上i位有连续多少个相同元素
这样子问题就能描述清楚了 
*/
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
#include<cmath>
using namespace std;
#define maxn 5009
#define inf 40000000
typedef pair<int,int >P;
const long long mo=1e9+7;
long long dp[maxn][2][100];
int n,a,b;
int main()
{
	int T;
	//printf("%lld\n",mo*1000);
	cin>>T;
	
	while(T--)
	{
		memset(dp,0,sizeof(dp));
		scanf("%d%d%d",&n,&a,&b);
		dp[1][0][1]=5;
		dp[1][1][1]=21;
		long long ans=26;
		for(int i=2;i<=n;i++)
		{
			//i位=0,有1个0;(即前面全是1) 
			for(int k=1;k<=min(b,i-1);k++)
			{
				dp[i][0][1]+=dp[i-1][1][k];
			}
			dp[i][0][1]=(dp[i][0][1]*5%mo);
			ans+=dp[i][0][1];
			//i位=0,有k(k>=2)个0;
			for(int k=2;k<=min(a,i);k++)
			{
				dp[i][0][k]=(dp[i-1][0][k-1]*5%mo);
				ans+=dp[i][0][k];
			}
			//i位=1,有1个1;(即前面全是0) 
			for(int k=1;k<=min(a,i-1);k++)
			{
				dp[i][1][1]+=dp[i-1][0][k];
			}
			dp[i][1][1]=(dp[i][1][1]*21%mo);
			ans+=dp[i][1][1];
			//i位=1,有k(k>=2)个1;
			for(int k=2;k<=min(b,i);k++)
			{
				dp[i][1][k]=(dp[i-1][1][k-1]*21%mo);
				ans+=dp[i][1][k];
			}
			ans%=mo;
		}
		printf("%lld\n",ans);
	}
	
	return 0;
}