375 Guess Number Higher or Lower II

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.

Hint:

  • The best strategy to play the game is to minimize the maximum loss you could possibly face. Another strategy is to minimize the expected loss. Here, we are interested in the first scenario.
  • Take a small example (n = 3). What do you end up paying in the worst case?
  • As a follow-up, how would you modify your code to solve the problem of minimizing the expected loss, instead of the worst-case loss?
class Solution {
public:
    int getMoneyAmount(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        // res[i][j]: result of money(i+1, j+1)
        // 1. res[i][i] = 0;
        // 2. res[i][i+1] = i+1;
        for ( int i = 0; i < n-1; i++ ) res[i][i+1] = i+1;
        for ( int d = 2; d <= n-1; d++ ) {
            for ( int i = 0; i < n-d; i++ ) {
                // res[i][i+d]
                int tmp = res[i+1][i+d] + i+1;
                tmp = min(tmp, res[i][i+d-1] + i+d+1);
                for ( int j = i+1; j < i+d; j++ ) tmp = min(tmp, j+1 + max(res[i][j-1], res[j+1][i+d]));
                res[i][i+d] = tmp;
            }
        }
        return res[0][n-1];
    }
};

Notes
和大楼扔鸡蛋那道题非常像!

results matching ""

    No results matching ""