题目大意
我们正在玩一个猜数游戏,游戏规则如下:
我从 1 到 n 之间选择一个数字,你来猜我选了哪个数字。
每次你猜错了,我都会告诉你,我选的数字比你的大了或者小了。
然而,当你猜了数字 x 并且猜错了的时候,你需要支付金额为 x 的现金。直到你猜到我选的数字,你才算赢得了这个游戏。
示例:
n = 10, 我选择了8.
第一轮: 你猜我选择的数字是5,我会告诉你,我的数字更大一些,然后你需要支付5块。
第二轮: 你猜是7,我告诉你,我的数字更大一些,你支付7块。
第三轮: 你猜是9,我告诉你,我的数字更小一些,你支付9块。
游戏结束。8 就是我选的数字。
你最终要支付 5 + 7 + 9 = 21 块钱。
给定 n ≥ 1,计算你至少需要拥有多少现金才能确保你能赢得这个游戏。
解题思路
首先,这道题并不适合用二分法来做,二分法只能减少猜的次数,并不能保证减少金钱。例如n=5,若用二分法,猜测目标是target=5,首先猜测x=3(猜错),然后再猜x=4(猜错),再猜x=5(猜对),用需要7块钱。
这道题应该考虑的是,在范围i~j
内任意猜测一个数字,至少需要多少钱才能够用。既然是任意猜测,至少需要,那我们就应该考虑最坏情况(对应任意)下需要的最少(对应至少)的钱。
假设在i~j
范围内,我们猜测数字k。既然我们要考虑最坏情况,就应当认为数字k不是我们要求的数字。数字k将范围分成了两部分,即i~k-1
和k+1~j
。我们要求的数字肯定是在其中一部分(题目说了会告诉你猜测的数字和要求数字的大小关系,因此我们只用在其中一个范围找就可以了)。
既然target在两个范围内的其中之一,且我们需要找到最坏情况,因此只要计算在范围i-k-1
和范围k+1~j
找到某个数至少需要多少钱,取二者最大值即可。这样我们就得到了猜测k时,在范围i~j
内的最坏情况。
另一方面,我们要考虑最坏情况下需要的最少钱,因此在范围i~j
内遍历所有猜测点k,求其中最小值即可(也就是找到最坏情况下的最少钱)。
class Solution {
public:
int getMoneyAmount(int n) {
if (n == 1)
return 0;
// 这里n+1纯粹是为了方面,因为范围是1~n,而索引是从0开始
vector<vector<int>> dp(n + 1, vector<int>(n + 1, 0));
for (int i = n; i >= 1; --i){
for (int j = i + 1; j <= n; ++j){
// 计算范围i~j内最坏情况下的最少钱
dp[i][j] = INT_MAX;
// 在猜测k时的最坏情况
// min()是求所有最坏情况中的最小值
for (int k = i; k <= j; ++k){
// k是左边界,则找右边的范围即可
if (k == i){
dp[i][j] = min(dp[i][j], dp[k + 1][j] + k);
}
else if (k == j){
// k是右边界,则找左边的范围即可
dp[i][j] = min(dp[i][j], dp[i][k - 1] + k);
}
else{
// k分出了两个子范围且target只能在一边,我们又要k时的最坏情况,因此找两个范围最大值
dp[i][j] = min(dp[i][j], max(dp[i][k - 1], dp[k + 1][j]) + k);
}
}
}
}
return dp[1][n];
}
};