LeetCode 877. Stone Game（数学 / dp）

题目链接

https://leetcode.com/problems/stone-game/

题意

一共有 n 堆石子排成一排（n 为偶数），每堆石子个数为 a[i] 个（保证石子总数为基数）。甲乙依次拿取石子，每次只能拿最左或最右的一堆，且两人都采取最优策略，最后获得石子数多的赢。甲为先手，若甲赢则输出 true，否则输出 false。

思路1

动态规划。设 dp[i][j] 为石子为第 i 堆至第 j 堆时，当前出手人最多可以比对手多 dp[i][j] 个石子。考虑：

• 只有 1 堆石子时（即 i = j），dp[i][j] = a[i]
• 只有 2 堆石子时，若取第 i 堆，则对手会获得 dp[i+1][j] 的收益，此时自己得到的收益为 a[i] - dp[i+1][j]；若取第 j 堆，则对手会获得 dp[i][j-1] 的收益，此时自己得到的收益为 a[j] - dp[i][j-1]。
• 因此递推方程为：

dp[i][j] = max(a[i] - dp[i+1][j], a[j] - dp[i][j-1]);

• 最后甲最多能比乙多的石子数为 dp[0][n-1]，若该数字大于 0 则甲胜。

代码

/**
 * @param {number[]} piles
 * @return {boolean}
 */
var stoneGame = function(piles) {
    let len = piles.length;
    let dp = [];
    for (let i = 0; i < len; i++) {
        let tmp = [];
        for (let j = 0; j < len; j++) {
            tmp.push(0);
        }
        dp.push(tmp);
    }
    // d 为假设当前有 d 堆石子
    for (let d = 0; d < len; d++) {
        dp[d][d] = piles[d];
        // 当前最左边的是第 s 堆石子
        for (let s = 0; s < len - d; s++) {
        	// 这里需要注意一下边界问题
            if (s + 1 >= len) {
                dp[s][s + d] = piles[s + d] - dp[s][s + d - 1];
            } else if (s + d - 1 < 0) {
                dp[s][s + d] = piles[s] - dp[s + 1][s + d];
            } else {
                dp[s][s + d] = Math.max(piles[s] - dp[s + 1][s + d], piles[s + d] - dp[s][s + d - 1]);
            }
        }
    }
    return dp[0][len - 1] > 0;
};

思路2

先手是可以保证自己只拿奇数堆，或只拿偶数堆的。
比如：现在一共 6 堆石子。

• 若先手想要只拿奇数堆，他先拿第 1 堆，后手只能拿第 2 或 6 堆。若后手拿第 2 堆，则先手拿第 3 堆，后手只能拿第 4 或 6 堆；若后手拿第 6 堆，则先手拿第 5 堆，后手又只能拿第 2 或 4 堆。如此以往，可以保证先手一定可以拿到 1、3、5 堆。
• 若先手想要只拿偶数堆，他只需要先拿第 6 堆，反复同上，可以保证先手一定拿到 2、4、6 堆。
• 因此先手只需要在第一回合就计算好，奇数堆的总数和偶数堆的总数哪个大即可保证自己必赢

代码

/**
 * @param {number[]} piles
 * @return {boolean}
 */
var stoneGame = function(piles) {
	return true;
}