[LintCode] Coins in a Line II

There are n coins with different value in a line. Two players take turns to take one or two coins from left side until there are no more coins left. The player who take the coins with the most value wins.

Could you please decide the first player will win or lose?

Example

Given values array A = [1,2,2], return true.

Given A = [1,2,4], return false.

 

Solution 1. Recursion 

In order for the first player to win, the coin value he gets must be bigger than a half of the total value of n coins. 

For a given start index that the first player can pick one or two coins at,  he can either pick one coin of values[startIdx] or two coins of values[startIdx] + values[startIdx + 1].  

Given that the second player plays optimally too, we have the following optimal substructure.

As we can see from the optimal substructure, there exists many overlapping subproblems that are redundantly recomputed, making this recursive solution inefficient.

26         int pickOneVal = Math.min(fPMaxValue(values, startIdx + 2, currVal), 
27                          fPMaxValue(values, startIdx + 3, currVal)) + values[startIdx];
28         int pickTwoVal = Math.min(fPMaxValue(values, startIdx + 3, currVal),
29                          fPMaxValue(values, startIdx + 4, currVal)) + values[startIdx] + values[startIdx + 1];
30         return Math.max(pickOneVal, pickTwoVal);

public class Solution {
    public boolean firstWillWin(int[] values) {
        if(values == null || values.length == 0){
            return false;
        }
        if(values.length <= 2){
            return true;
        }
        int sum = 0; 
        for(int i = 0; i < values.length; i++){
            sum += values[i];
        }
        return fPMaxValue(values, 0, 0) > sum / 2;
    }
    private int fPMaxValue(int[] values, int startIdx, int currVal){
        int diff = values.length - startIdx;
        if(diff == 2 || diff == 3){
            return currVal + values[startIdx] + values[startIdx + 1];
        }
        if(diff == 1){
            return currVal + values[startIdx];
        }
        if(diff <= 0){
            return currVal;
        }
        int pickOneVal = Math.min(fPMaxValue(values, startIdx + 2, currVal), 
                         fPMaxValue(values, startIdx + 3, currVal)) + values[startIdx];
        int pickTwoVal = Math.min(fPMaxValue(values, startIdx + 3, currVal),
                         fPMaxValue(values, startIdx + 4, currVal)) + values[startIdx] + values[startIdx + 1];
        return Math.max(pickOneVal, pickTwoVal);
    }
}

 

Solution 2. Dynamic Programming 

Since solution 1 has both optimal substructure and overlapping subproblems, we can apply dynamic programming to avoid redundant recomputation of subproblems.

Dp state: dp[i] is the most value player one can get when there is i coins left.

Dp function: 

dp[i] = Math.max(Math.min(dp[i - 2], dp[i - 3]) + values[n - i],  Math.min(dp[i - 3], dp[i - 4]) + values[n - i] + values[n - i + 1]);

 

public class Solution {
    public boolean firstWillWin(int[] values) {
        if(values == null || values.length == 0){
            return false;
        }
        if(values.length <= 2){
            return true;
        }
        int sum = 0; 
        for(int i = 0; i < values.length; i++){
            sum += values[i];
        }
        int n = values.length;
        int[] dp = new int[n + 1];
        dp[0] = 0;
        dp[1] = values[n - 1];
        dp[2] = values[n - 2] + values[n - 1];
        dp[3] = values[n - 3] + values[n - 2];
        for(int i = 4; i <= n; i++){
            dp[i] = Math.max(Math.min(dp[i - 2], dp[i - 3]) + values[n - i], 
                             Math.min(dp[i - 3], dp[i - 4]) + values[n - i] + values[n - i + 1]);
        }
        return dp[n] > sum / 2;
    }
}