877. 石头游戏

877. 石头游戏

亚历克斯和李用几堆石子在做游戏。偶数堆石子排成一行，每堆都有正整数颗石子 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 。

提示：

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

使用动态规划算法求解

状态包括：

• 石头序列 piles[i…j] 开始索引 i
• 石头序列 piles[i…j] 结尾索引 j
• 取石头的人 k （k 的取值范围是 {0, 1}，0 代表先手 1 代表后手）

dp[i][j][k] 表示对于 piles[i…j] 这一堆石头， k 能拿到的最高分数。

k = 0 表示先手取石子，可以取左边（得到 piles[i]，然后面对 piles[i+1…j] 时作为后手），可以取右边（得到 piles[j]，然后面对 piles[i…j-1] 时作为后手）。

dp[i][j][0] = max(dp[i+1][j][1] + piles[i], dp[i][j-1][1] + piles[j])

若先手取左边，后手面对 piles[i+1…j] 时作为先手；

dp[i][j][1] = dp[i+1][j][0]

若先手取右边，后手面对 piles[i…j-1] 时作为先手；

dp[i][j][1] = dp[i][j-1][0]

初始情况：

i == j，即只有一堆石子，先手拿走全部 dp[i][j][0] = piles[i], dp[i][j][1] = 0

推导 dp[i][j][k] 需要知道 dp[i+1][j][k] 和 dp[i][j-1][k]， dp矩阵中的相对位置如下，需要斜着遍历数组。
$$\begin{array}{cc}dp[k{]}_{i,j-1}& dp[k{]}_{i,j}\\ dp[k{]}_{i+1,j-1}& dp[k{]}_{i+1,j}\end{array}$$

class Solution:
    def stoneGame(self, piles: List[int]) -> bool:
        piles_size = len(piles)

        # 初始化 dp
        dp = [[[0, 0] if i != j else [piles[i], 0] for j in range(piles_size)] for i in range(piles_size)]

        # 斜着遍历
        for t in range(piles_size):
            for i in range(piles_size-t):
                j = i + t
                if i == j: continue
                left = dp[i+1][j][1] + piles[i]
                right = dp[i][j-1][1] + piles[j]
                if left > right:
                    dp[i][j][0] = left
                    dp[i][j][1] = dp[i+1][j][0]
                else:
                    dp[i][j][0] = right
                    dp[i][j][1] = dp[i][j-1][0]
        
        return True if dp[0][piles_size-1][0] > dp[0][piles_size-1][1] else False