LeetCode 1140. Stone Game II

原题链接在这里：https://leetcode.com/problems/stone-game-ii/

题目：

Alex and Lee continue their games with piles of stones.  There are a number of piles arranged in a row, and each pile has a positive integer number of stones piles[i].  The objective of the game is to end with the most stones. 

Alex and Lee take turns, with Alex starting first.  Initially, M = 1.

On each player's turn, that player can take all the stones in the first X remaining piles, where 1 <= X <= 2M.  Then, we set M = max(M, X).

The game continues until all the stones have been taken.

Assuming Alex and Lee play optimally, return the maximum number of stones Alex can get.

Example 1:

Input: piles = [2,7,9,4,4]
Output: 10
Explanation:  If Alex takes one pile at the beginning, Lee takes two piles, then Alex takes 2 piles again. Alex can get 2 + 4 + 4 = 10 piles in total. If Alex takes two piles at the beginning, then Lee can take all three piles left. In this case, Alex get 2 + 7 = 9 piles in total. So we return 10 since it's larger. 

Constraints:

• 1 <= piles.length <= 100
• 1 <= piles[i] <= 10 ^ 4

题解：

For the current play A, if 2*M could reach the end of piles array, then A get them all. 

Otherwise, A gets all - the minimum player could get. Cash value hash[i][M], which means max value from current i with value of M, in case of duplicate calculation.

Time Complexity: O(2^n). dfs takes time T(n). T(n) = O(1) + 2*M*T(n-x).

Space: O(n^2).

AC Java:

class Solution {
    int [] sum;
    int [][] hash;
    
    public int stoneGameII(int[] piles) {
        int n = piles.length;
        sum = new int[n];
        sum[n-1] = piles[n-1];
        for(int i = n-2; i>=0; i--){
            sum[i] = sum[i+1] + piles[i];
        }
        
        hash = new int[n][n];
        
        return dfs(piles, 0, 1);
    }
    
    private int dfs(int [] piles, int i, int M){
        if(i+2*M >= piles.length){
            return sum[i];
        }
        
        if(hash[i][M] != 0){
            return hash[i][M];
        }
        
        int min = Integer.MAX_VALUE;
        for(int x = 1; x<=2*M; x++){
            min = Math.min(min, dfs(piles, i+x, Math.max(M, x)));
        }
        
        // all the left stones - minimum stones the other player could get
        hash[i][M] = sum[i]-min;
        return hash[i][M];
    }
}

类似877. Stone Game.