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.
The problem can be done by DFS + memorization.
Note that the problem asks for the worst case scenario since you need to guarantee a win. In the worst case scenario, you need to guess wrong for each time expect for there is only 1 number.
Code (Java):
class Solution {
public int getMoneyAmount(int n) {
int[][] dp = new int[n + 1][n + 1];
getMoneyAmountHelper(1, n, dp);
return dp[1][n];
}
private int getMoneyAmountHelper(int start, int end, int[][] dp) {
if (start >= end) {
return 0;
}
if (dp[start][end] != 0) {
return dp[start][end];
}
int minCost = Integer.MAX_VALUE;
for (int i = start; i <= end; i++) {
int cost = i + Math.max(getMoneyAmountHelper(start, i - 1, dp),
getMoneyAmountHelper(i + 1, end, dp));
minCost = Math.min(minCost, cost);
}
dp[start][end] = minCost;
return minCost;
}
}
No comments:
Post a Comment