Java 算法-硬币排成线II(动态规划)

  好久没有做面试题了,感觉自己的编程能力又变弱了。
  今天在lintCode上做了一道题,关于动态规划,比较有意思。

题意:
有 n 个不同价值的硬币排成一条线。两个参赛者轮流从左边依次拿走 1 或 2 个
硬币,直到没有硬币为止。计算两个人分别拿到的硬币总价值,价值高的人获
胜。

请判定 第一个玩家 是输还是赢?
样例:
给定数组 A = [1,2,2], 返回 true.

给定数组 A = [1,2,4], 返回 false.

1.解题思路

  这道题我们从动态规划的角度上分析。
  首先,如果只有1个或者2个硬币的话,肯定是第一个玩家赢,这个是不可置疑的。
  一个数组,我们从后往前分析(我们假设硬币的下标是i, len为数组的长度),定义一个dp数组,从i到len的最大值。
  1.假设 i = len,表示当前没有硬币了,所以 dp[len] = 0;
  2.假设 i = len - 1,表示当前还有一个硬币能够拿,为了保证最大,肯定会拿,所以dp[len - 1] = values[len - 1];
  3.假设 i = len - 2,表示当前还有两个硬币能够拿,为了保证最大,所以两个都拿,dp[len - 2] = values[len - 2] + values[len - 1]
  4.假设 i = len - 3,表示当前还有三个硬币能够拿,为了保证最大,所以两个都拿(因为只拿一个的话,另一个人肯定两个,有可能比我们的大),dp[len - 2] = values[len - 2] + values[len - 1]
  5.假设 i = len -4以及更小,表示当前还有四个或者更多硬币可以拿,此时分为两种情况:
    A.只拿一个硬币,也就是说,当前,我们拿len - 4那个硬币,但是没说不拿之后(那个是当前i之后的操作了,我们这里是倒着推导的,所以i之后的都被计算了),只是说当前只拿len - 4。这里涉及到另一个的拿法,另一个肯定会想法设法的拿最多,它可能拿一个或者两个,当是拿一个时,我们需要计算dp[i + 2],两个时,我们需要计算dp[i + 3],从而使得我们最少(最小),所以此时dp[i] = values[i] + Math.min(dp[i + 2], dp[i + 3])。
    B.拿两个硬币,就是说拿len -3 和len -2两个位置上的硬币。 同样的道理,dp[i] = values[i] + values[i + 1] + Math.min(dp[i + 3], dp[i + 4]);
  最后我们在取两者的最大值就行

2.代码

public boolean firstWillWin(int[] values) {
        if (values.length == 0) {
            return false;
        }
        if (values.length <= 2) {
            return true;
        }
        int len = values.length;
        int dp[] = new int[len + 1];
        dp[len] = 0;  //没有硬币可以拿
        dp[len - 1] = values[len - 1]; //只有一个可以拿,肯定拿
        dp[len - 2] = values[len - 2] + values[len - 1];  //只有两个可以拿,都拿
        dp[len - 3] = values[len - 3] + values[len - 2];  //只有三个可以拿,拿前两个,因为保证最大
        for (int i = len - 4; i >= 0; i--) {
            dp[i] = Math.max(values[i] + Math.min(dp[i + 2], dp[i + 3]),
                    values[i + 1] + values[i + 2] + Math.min(dp[i + 3], dp[i + 4]));
        }
        int sum = 0;
        for(int i = 0; i < values.length; i++) {
            sum += values[i];
        }
        int a = sum - dp[0];
        return dp[0] > a;
    }

你可能感兴趣的:(Java 算法-硬币排成线II(动态规划))