[leetcode]Guess Number Higher or Lower II
链接:https://leetcode.com/problems/guess-number-higher-or-lower-ii/description/
Question
We are playing the Guess Game. The game is as follows:
I pick a number from 1 to n. You have to guess which number I picked.
Every time you guess wrong, I’ll tell you whether the number I picked is higher or lower.
However, when you guess a particular number x, and you guess wrong, you pay $x. You win the game when you guess the number I picked.
Example
n = 10, I pick 8.
First round: You guess 5, I tell you that it's higher. You pay $5.
Second round: You guess 7, I tell you that it's higher. You pay $7.
Third round: You guess 9, I tell you that it's lower. You pay $9.
Game over. 8 is the number I picked.
You end up paying $5 + $7 + $9 = $21.
Given a particular n ≥ 1, find out how much money you need to have to guarantee a win.
Solution
class Solution {
public:
#define inf 0x3fffffff
int getMoneyAmount(int n) {
vector<vector<int> > dp;
dp.resize(n+1);
for (int i = 0; i < n+1; i++) dp[i].resize(n+1, 0);
for (int i = 1; i <= n; i++) dp[i][i] = 0;
for (int i = 1; i <= n-1; i++) dp[i][i+1] = i;
for (int i = 1; i <= n-2; i++) dp[i][i+2] = i+1;
for (int len = 3; len <= n-1; len++) {
for (int i = 1; i <= n-len; i++) {
int min_val = inf;
for (int j = i; j <= i+len; j++) {
if (j == i) min_val = min(min_val, j+dp[j+1][i+len]);
else if (j == i+len) min_val = min(min_val, dp[i][i+len-1]+j);
else min_val = min(min_val, j+max(dp[i][j-1], dp[j+1][i+len]));
}
dp[i][i+len] = min_val;
}
}
return dp[1][n];
}
};
思路:当区间只有1个数,不需要pay;当区间只有两个数,肯定猜小的数划算;当区间有3个数,猜中间的划算。所以这里的dp[i][j]表示从i到j区间时候最多的pay是多少(当然我不是傻子,肯定是最优情况下的最多pay)。