蓝桥杯2022年第十三届省赛真题-李白打酒加强版

原题链接

蓝桥杯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];
}