题目:预测赢家
给定一个表示分数的非负整数数组。 玩家 1 从数组任意一端拿取一个分数,随后玩家 2 继续从剩余数组任意一端拿取分数,然后玩家 1 拿,…… 。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数总和最多的玩家获胜。
给定一个表示分数的数组,预测玩家1是否会成为赢家。你可以假设每个玩家的玩法都会使他的分数最大化。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/predict-the-winner
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例 1:
输入:[1, 5, 2]
输出:False
解释:一开始,玩家1可以从1和2中进行选择。
如果他选择 2(或者 1 ),那么玩家 2 可以从 1(或者 2 )和 5 中进行选择。如果玩家 2 选择了 5 ,那么玩家 1 则只剩下 1(或者 2 )可选。
所以,玩家 1 的最终分数为 1 + 2 = 3,而玩家 2 为 5 。
因此,玩家 1 永远不会成为赢家,返回 False 。
示例 2:
输入:[1, 5, 233, 7]
输出:True
解释:玩家 1 一开始选择 1 。然后玩家 2 必须从 5 和 7 中进行选择。无论玩家 2 选择了哪个,玩家 1 都可以选择 233 。
最终,玩家 1(234 分)比玩家 2(12 分)获得更多的分数,所以返回 True,表示玩家 1 可以成为赢家。
解题思路
- 定义状态:用 d p [ i ] [ j ] dp[i][j] dp[i][j] 表示玩家1与玩家二的相对分数。
- 状态转移方程: d p [ i ] [ j ] = m a x ( n u m s [ i ] − d p [ i + 1 ] [ j ] , n u m s [ j ] − d p [ i ] [ j − 1 ] ) dp[i][j] = max(nums[i] - dp[i + 1][j], nums[j] - dp[i][j - 1]) dp[i][j]=max(nums[i]−dp[i+1][j],nums[j]−dp[i][j−1])。
- 通过状态方程可以看出状态依赖于后面的 i + 1 i+1 i+1 和 j − 1 j-1 j−1,所以得让 i i i 从大向小变换,同时 j j j 从小到大。
- 临界情况判断:区间取值: [ i , j ] [i, j] [i,j],需要初始化第 n − 1 n-1 n−1 行,为了保证 j + 1 j+1 j+1 一定计算过,则需要将所有的 d p [ i ] [ j ] dp[i][j] dp[i][j] 初始化为 n u m s [ i ] nums[i] nums[i]。
代码
class Solution {
public:
bool PredictTheWinner(vector<int>& nums) {
//dp[i][j] = max(nums[i] - dp[i + 1][j], nums[j] - dp[i][j - 1])
//通过状态方程可以看出状态依赖于后面的i+1和j-1,所以得让i从大向小变换,同时j从小到大
int n = nums.size();
vector<vector<int>> dp(n, vector<int>(n));
//初始化第n-1行
//将所有的dp[i][i]置为nums[i]
for(int i = 0; i < n; ++i){
dp[i][i] = nums[i];
}
for(int i = n - 2; i >= 0; --i){
for(int j = i + 1; j < n; ++j){
dp[i][j] = max(nums[i] - dp[i + 1][j], nums[j] - dp[i][j - 1]);
}
}
return dp[0][n - 1] >= 0;
}
};