Leetcode之动态规划（DP）专题-877. 石子游戏（Stone Game）

亚历克斯和李用几堆石子在做游戏。偶数堆石子排成一行，每堆都有正整数颗石子 piles[i] 。

游戏以谁手中的石子最多来决出胜负。石子的总数是奇数，所以没有平局。

亚历克斯和李轮流进行，亚历克斯先开始。每回合，玩家从行的开始或结束处取走整堆石头。这种情况一直持续到没有更多的石子堆为止，此时手中石子最多的玩家获胜。

假设亚历克斯和李都发挥出最佳水平，当亚历克斯赢得比赛时返回 true ，当李赢得比赛时返回 false 。

示例：

输入：[5,3,4,5]
输出：true
解释：
亚历克斯先开始，只能拿前 5 颗或后 5 颗石子 。
假设他取了前 5 颗，这一行就变成了 [3,4,5] 。
如果李拿走前 3 颗，那么剩下的是 [4,5]，亚历克斯拿走后 5 颗赢得 10 分。
如果李拿走后 5 颗，那么剩下的是 [3,4]，亚历克斯拿走后 4 颗赢得 9 分。
这表明，取前 5 颗石子对亚历克斯来说是一个胜利的举动，所以我们返回 true 。

提示：

2 <= piles.length <= 500
piles.length 是偶数。
1 <= piles[i] <= 500
sum(piles) 是奇数。

数学题，但我们用DP来求解这一题。

我们首先定义一个类，名为P：

private static class P {
        int fir, sec;

        P(int fir, int sec) {
            this.fir = fir;
            this.sec = sec;
        }
    }

P中有两个属性，fir代表先手获得的最高分数，sec代表后手获得的最高分数

那么我们就可以写出DP的含义

我们把dp定义为2维，如dp[i][j]代表从这堆石子的第i堆选到第j堆中获得的最高分数。

那么dp[i][j].fir 表示从第i堆到第j堆中先手获得的最高分。

dp[i][j].sec 表示从第i堆到第j堆中后手获得的最高分。

例如：piles = [5，3，4，5]

dp[0][1].fir = 5, 面对[5,3]，先手可以得5分

dp[0][1].sec = 3，面对[5,3],后手可以得3分

我们首先明确一点，当 i == j 时，即只有一堆石子的时候，上面的例子，例如 i == j == 0 那么 dp[i][j].fir = 5 sec=0

所以我们把i==j时的所有情况，遍历一遍。

for (int i = 0; i < n; i++) {
            dp[i][i].fir = piles[i];
            dp[i][i].sec = 0;
}

下面看一张图，是最终dp数组的最后状态：

我们可以看到，dp[0][1] = （9，3）即dp[0][1].fir = 9 sec=3

这个数值是由（0，0）和（1，1）一起生成的。

我们可以按层遍历，即：

第1层：对角线

第2层：（9，3）（9，1）（2，1）

第3层：（4，9）（10，2）

第4层：（11，4）

选择了按层遍历后，我们需要得到状态转移方程：

面对一堆石头，我们有如下情况可以选择：

1、我是先手

我选左边，面对剩下的piles[i+1,j]
我选右边，面对剩下的piles[i,j-1]

随后对方变成先手，我变成后手

dp[i][j].fir = max(piles[i]+dp[i+1][j].sec,piles[j]+dp[i][j-1].sec);

2、我是后手

先手选择了左边的那堆，我只能选择剩下的，dp[i][j].sec = dp[i+1][j].fir;
先手选择了右边的那堆，我只能选择剩下的，dp[i][j].sec = dp[i][j-1].fir;

随后对方变成后手，我变成了先手

class Solution {
    private static class P {
        int fir, sec;

        P(int fir, int sec) {
            this.fir = fir;
            this.sec = sec;
        }
    }

    public boolean stoneGame(int[] piles) {

        int n = piles.length;
        P[][] dp = new P[n + 1][n + 1];
        for (int i = 0; i < n; i++) {
            for (int j = i; j < n; j++) {
                dp[i][j] = new P(0, 0);
            }
        }
        for (int i = 0; i < n; i++) {
            dp[i][i].fir = piles[i];
            dp[i][i].sec = 0;
        }

        for (int c = 2; c <= n; c++) {
            for (int i = 0; i <= n - c; i++) {
                int j = c + i - 1;
                int left = piles[i] + dp[i + 1][j].sec;
                int right = piles[j] + dp[i][j - 1].sec;
                if (left > right) {
                    dp[i][j].fir = left;
                    dp[i][j].sec = dp[i + 1][j].fir;
                } else {
                    dp[i][j].fir = right;
                    dp[i][j].sec = dp[i][j - 1].fir;
                }

            }
        }
        return dp[0][n - 1].fir > dp[0][n - 1].sec;
    }
}

#本题思路来自题解区

Leetcode之动态规划（DP）专题-877. 石子游戏（Stone Game）

猜你喜欢