题目大意
亚历克斯和李用几堆石子在做游戏。偶数堆石子排成一行,每堆都有正整数颗石子 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 。
解题思路
我们只计算自己能够得到多少分。首先有两种选择,从左边拿或者从右边拿。对于每种选择,对方将会从剩下的石子中选取。因为双方都是理性的,对方肯定会使自己得分最大化,反之就是让你的得分最小化。因此有两种取法,会选择一种你得分最小的取法然你继续。
class Solution {
public:
bool stoneGame(vector<int>& piles) {
int socre = myScore(piles, 0, piles.size() - 1);
int total = accumulate(piles.begin(), piles.end(), 0);
return socre > total - socre;
}
int myScore(vector<int> & piles, int left, int right){
if (left > right)
return 0;
// 你取左边的石子,对方取一颗后,会返回一种你得分最低的情况
// 所以你只能从对方取完的两种情况中选择得分低的情况继续
int leftScore = piles[left] + min(myScore(piles, left + 2, right), myScore(piles, left + 1, right - 1));
// 取右边的石子同理
int rightScore = piles[right] + min(myScore(piles, left + 1, right - 1), myScore(piles, left, right - 2));
// 返回你自己的两种情况的最大值作为你的得分
return max(leftScore, rightScore);
}
};
将上面的递归转化成动态规划。
class Solution {
public:
bool stoneGame(vector<int>& piles) {
int length = piles.size();
vector<vector<int>> dp(length, vector<int>(length, 0));
for(int i = length - 1; i >= 0; --i){
for (int j = i; j < length; ++j){
if (i == j)
dp[i][j] = piles[i];
else if (i + 1 == j)
dp[i][j] = max(piles[i], piles[j]);
else
dp[i][j] = max(piles[i] + min(dp[i + 2][j], dp[i + 1][j - 1]), piles[j] + min(dp[i + 1][j - 1], dp[i][j - 2]));
}
}
return dp[0].back();
}
};