【Lintcode】395. Coins in a Line II

题目地址：

https://www.lintcode.com/problem/coins-in-a-line-ii/description

有$n$个不同价值的硬币排成一条线，硬币价值以数组$A$表示。两个参赛者轮流从左边依次拿走$1$或$2$个硬币，直到没有硬币为止。计算两个人分别拿到的硬币总价值，价值高的人获胜。问先手是否有必胜策略。

思路是动态规划。设$f[i]$为，如果只剩下$A[i:l_A-1]$这些硬币的时候，先手最多能赢后手多少价值（意思是先手与后手分差最大是多少）。那么$f[i]$就分两种情况，第一种是先手只拿$A[i]$，这样后手最多能拿到的价值就是$f[i+1]$，这样分差就是$A[i]-f[i+1]$；类似地，第二种就是先手拿$A[i]$和$A[i+1]$，则分差就是$A[i]+A[i+1]-f[i+2]$。边界情况是$f[l_A-1]=A[l_A-1],f[l_A-2]=A[l_A-2]+A[l_A-1]$，递推式是： f [ i ] = max ⁡ { A [ i ] + A [ i + 1 ] − f [ i + 2 ] , A [ i ] − f [ i + 1 ] } f[i]=\max\{A[i]+A[i+1]-f[i+2],A[i]-f[i+1]\} f[i]=max{A[i]+A[i+1]−f[i+2],A[i]−f[i+1]}最后只需要返回$f[0]$是否大于$0$即可。代码如下：

public class Solution {
    /**
     * @param values: a vector of integers
     * @return: a boolean which equals to true if the first player will win
     */
    public boolean firstWillWin(int[] values) {
        // write your code here
        int[] dp = new int[values.length];
        for (int i = values.length - 1; i >= 0; i--) {
            if (i == values.length - 1){
                dp[i] = values[i];
            } else if (i == values.length - 2) {
                dp[i] = values[i] + values[i + 1];
            } else {
                dp[i] = Math.max(values[i] + values[i + 1] - dp[i + 2], values[i] - dp[i + 1]);
            }
        }
        
        return dp[0] > 0;
    }
}

时空复杂度$O(n)$。