【P8786 [蓝桥杯 2022 省 B] 李白打酒加强版】

题解：蓝桥杯 2022 省赛 B 组 I 题

题目描述

大诗人李白，以他的豪爽和好酒闻名。在一天的酒途中，他提着酒壶，壶中有 $2$ 斗美酒，一路上遇到了店和花。他边走边唱：

无事街上走，提壶去打酒。
逢店加一倍，遇花喝一斗。

这一路上，他一共遇到了店 $N$ 次，遇到了花 $M$ 次。问题是，最后一次他遇到的是花，而且刚好把酒喝光了。现在的问题是：有多少种不同的可能的店和花的遇到顺序？

需要注意的是，壶里没酒（$0$ 斗）时遇店是合法的，加倍后还是没酒；但是没酒时遇花是不合法的。

输入格式

第一行包含两个整数 $N$ 和 $M$。

输出格式

输出一个整数表示答案。由于答案可能很大，输出模 $1000000007$（即 $10^9+7$）的结果。

样例输入和输出

输入：

5 10

输出：

14

解题思路

动态规划解决。使用三维数组 dp[i][j][k] 表示在遇到第 $i$ 个店和第 $j$ 个花，以及遇到 $k$ 个花的情况下的方案数。其中，$i$、$j$、$k$ 的范围分别是 $0\le i\le N$，$0\le j\le M$，$0\le k\le M$。

初始状态为 dp[0][0][2] = 1（表示初始状态下只有 2 斗酒），dp[0][1][1] = 1（表示初始状态下只有 1 斗酒，同时遇到了一朵花）。

使用三重循环来填充 dp 数组，根据题目中的条件进行状态转移：

1. 如果 $k$ 是偶数，那么 dp[i][j][k] = (dp[i - 1][j][k / 2] + dp[i][j - 1][k + 1]) % mod，表示在这一步的前一步可以选择遇到店或遇到花。
2. 如果 $k$ 是奇数，那么 dp[i][j][k] = dp[i][j - 1][k + 1]，表示只能遇到花，因为酒为奇数，表明上一步一定是遇到花才过来的。

最终答案就是 dp[N][M-1][1]，表示遇到了 $N$ 次店和 $M-1$ 次花，同时还有 1 斗酒的方案数。

完整代码

#include<bits/stdc++.h>

using namespace std;

int mod = 1e9 + 7;

long long dp[105][105][105] = {0}; 
//dp[i][j][k]表示第i个店第j个花，遇到k个花的方案数


int main() {
    
    int n,m;
    cin >> n >> m;
    dp[0][0][2] = 1; //只遇到花
    dp[0][1][1] = 1; //只遇到花
    dp[0][2][0] = 1; //只遇到花
    for (int i = 0; i <= 7; i++)
    {
        dp[i][0][int(pow(2, i + 1))] = 1;//只遇到店
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            for (int k = 0; k <= m; k++)
            {
                if (k % 2 == 0)
                    dp[i][j][k] = (dp[i - 1][j][k / 2] + dp[i][j - 1][k + 1]) % mod; 
                    //遇到酒店 或者 遇到 花
                else
                    dp[i][j][k] = dp[i][j - 1][k + 1];  
                    //只能遇到花

            }
        }
    }
    cout << dp[n][m-1][1] << endl;
    return 0;
}