2022 蓝桥杯李白打酒加强版 python

原创已于 2022-05-26 22:53:57 修改 · 621 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#动态规划 #算法

于 2022-05-26 22:53:28 首次发布

本文介绍了一个使用动态规划解决2022年蓝桥杯李白打酒问题的Python程序。通过三维数组dp记录状态转移过程，解决了李白在特定规则下买酒的问题。

2022 蓝桥杯李白打酒加强版 python

N, M = map(int,input().split())
dp = [[[0 for i in range(102)]for j in range(M+1)]for k in range(N+1)]
dp[0][0][2] = 1
dp[0][1][1] = 1
dp[1][0][4] = 1 
mod = 1000000007
for k in range(1,N+1):
    for j in range(M+1):
        for i in range(100, 0, -1):
            if i/2 % 1 == 0: 
                dp[k][j][i] = (dp[k-1][j][i//2] + dp[k][j-1][i+1]) % mod
            elif i/2 % 1 != 0:
                dp[k][j][i] = dp[k][j-1][i+1] % mod
        dp[k][j][0] = dp[k][j-1][1] % mod
print(dp[N][M][0])