LeetCode375. 猜数字大小 II(python,动态规划)

1. 题目

我们正在玩一个猜数游戏,游戏规则如下:

我从 1 到 n 之间选择一个数字,你来猜我选了哪个数字。

每次你猜错了,我都会告诉你,我选的数字比你的大了或者小了。

然而,当你猜了数字 x 并且猜错了的时候,你需要支付金额为 x 的现金。直到你猜到我选的数字,你才算赢得了这个游戏。

示例:

n = 10, 我选择了8.

第一轮: 你猜我选择的数字是5,我会告诉你,我的数字更大一些,然后你需要支付5块。
第二轮: 你猜是7,我告诉你,我的数字更大一些,你支付7块。
第三轮: 你猜是9,我告诉你,我的数字更小一些,你支付9块。

游戏结束。8 就是我选的数字。

你最终要支付 5 + 7 + 9 = 21 块钱。
给定 n ≥ 1,计算你至少需要拥有多少现金才能确保你能赢得这个游戏。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/guess-number-higher-or-lower-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解答

这里转移方程注意的是,猜了一个数字 k 后,只有一个方向继续猜,选择cost较大的那个方向。

class Solution:
    def getMoneyAmount(self, n: int) -> int:
        # dp[i][j] 猜中i-j用的最小钱
        dp = [[0 for j in range(n+1)] for i in range(n+1)]
        for d in range(1,n):
            for i in range(1,n+1):
                if i+d > n:
                    break
                min_cost = float('inf')
                for k in range(i,i+d+1):
                    cost = k + max(dp[i][k-1], (dp[k+1][i+d] if k+1<=i+d else 0))
                    min_cost = min(min_cost,cost)
                dp[i][i+d] = min_cost
        return dp[1][n]
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页