石子游戏
亚历克斯和李用几堆石子在做游戏。偶数堆石子排成一行,每堆都有正整数颗石子 piles[i]
.
游戏以谁手中的石子最多来决出胜负。石子的总数是奇数,所以没有平局。
亚历克斯和李轮流进行,亚历克斯先开始。 每回合,玩家从行的开始或结束处取走整堆石头。 这种情况一直持续到没有更多的石子堆为止,此时手中石子最多的玩家获胜。
- 亚历克斯先手,假设亚历克斯和李都发挥出最佳水平,求最后两个分别获得的石子个数
示例1
输入:[5,3,4,5]
输出:true
解释:
亚历克斯先开始,只能拿前 5 颗或后 5 颗石子 。
假设他取了前 5 颗,这一行就变成了 [3,4,5] 。
如果李拿走前 3 颗,那么剩下的是 [4,5],亚历克斯拿走后 5 颗赢得 10 分。
如果李拿走后 5 颗,那么剩下的是 [3,4],亚历克斯拿走后 4 颗赢得 9 分。
返回 [9, 8] 。
labuladong大佬详解 (opens new window)
提示
2 <= piles.length <= 500
piles.length
是偶数。1 <= piles[i] <= 500
sum(piles)
是奇数。
动态规划
- base
dp[i][j].fir = piles[i]
dp[i][j].sec = 0
- 其中
0 <= i == j < n
- 状态转移方程
dp[i][j].fir = max(piles[i] + dp[i+1][j].sec, piles[j] + dp[i][j-1].sec)
- 先手选择左边:
dp[i][j].sec = dp[i+1][j].fir
- 先手选择右边:
dp[i][j].sec = dp[i][j-1].fir
export const stoneGame = piles => {
const n = piles.length
const dp = []
// 初始化dp数组
for (let i = 0; i < n; i++) {
dp[i] = []
for (let j = i; j < n; j++) {
dp[i][j] = { fir: 0, sec: 0 }
}
}
// 填入base
for (let i = 0; i < n; i++)
dp[i][i].fir = piles[i]
// 倒着斜着遍历数组
for (let i = n - 2; i >= 0; i--) {
for (let j = i + 1; j < n; j++) {
// 先手选择最左边或最右边的分数
const left = piles[i] + dp[i + 1][j].sec
const 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 Object.values(dp[0][n - 1])
}
console结果可能不准确,按F12打开控制台查看
← 螺旋矩阵 实现 strStr() 函数。 →