原题链接
蓝桥杯2022年第十三届省赛真题-李白打酒加强版 - C语言网 (dotcpp.com)
题目描述
三维动态规划(满分AC)
#include <bits/stdc++.h>
using namespace std;
int N, M;
int mod = 1000000007;
// dp[剩余的店][剩余的花][剩余的酒]表示当前状态的累计可能情况数
long long dp[103][103][103] = {0};
int main()
{
cin >> N >> M;
// 从家里出来,酒壶中有酒 2 斗
dp[N][M - 1][1] = 1; // 第一次遇到了花,遇花喝一斗
dp[N - 1][M][4] = 1; // 第一次遇到了店,逢店加一倍
for (int i = N; i >= 0; i--) // N个店
{
for (int j = M; j >= 0; j--) // M个花
{
for (int k = j; k >= 0; k--) // 酒最多不超过花的数量
{
if (i == N && j == M)
{
break; // 要么遇到花,要么遇到店,不可能什么都没有遇到
}
// 最后一次遇到的是花, 他正好把酒喝光了。
// 所以当花<=1时,剩余的店必须为0;
if (j <= 1 && i != 0)
{
break;
}
if (dp[i][j][k] != 0)
{
continue; // 防止初始状态被置0(第一次遇到花或店)
}
// 状态转移方程
// 当前的情况总数 = 上一次遇到花的情况数 + 上一次遇到店的情况数
if (k % 2 == 0)
{
dp[i][j][k] = (dp[i][j + 1][k + 1] + dp[i + 1][j][k / 2]) % mod;
}
else
{
dp[i][j][k] = dp[i][j + 1][k + 1] % mod;
}
}
}
}
cout << dp[0][0][0];
}