思路1:暴力递归法
会超时
#include <iostream>
#include <vector>
using namespace std;
class Solution{
public:
// 为先手时
int first(int l,int r,vector<int> &nums){
// 只有一张牌,先手,直接返回该牌的值
if(l == r) return nums[l];
// 可以取第 l 张牌或者第 r 张牌,此时,这人为l+1到r或者l到r-1的后手
return max(nums[l]+second(l+1,r,nums),nums[r]+second(l,r-1,nums));
}
// 为后手时
int second(int l,int r,vector<int> &nums){
// 一张牌直接返回0,因为后手拿不到
if(l == r) return 0;
// 因为第一个人先拿,而且第一个人也足够聪明,所以第二个人拿到的只能是先拿是最小的那个
return min(first(l+1,r,nums),first(l,r-1,nums));
}
int getMaxScore(int n,vector<int> &nums){
if(n == 0) return 0;
// 返回A和B玩家能拿到的分数中最大那一个
return max(first(0,n-1,nums),second(0,n-1,nums));
}
};
int main(){
int n;
cin >> n;
vector<int> nums(n);
for(int i = 0;i < n;i++) cin >> nums[i];
Solution *s = new Solution();
int res = s->getMaxScore(n,nums);
cout << res << endl;
}
思路2:dp
将上面的递归,写成dp形式
#include <iostream>
#include <vector>
using namespace std;
class Solution{
public:
int getMaxScore(int n,vector<int> &nums){
if(n == 0) return 0;
// 先手时,[i][j]区间的最大分数
// 后手时,[i][j]区间的最大分数
vector<vector<int>> first(n,vector<int>(n,0));
vector<vector<int>> second(n,vector<int>(n,0));
// 这里也是从下往上,从左往右递归
for(int i = 0;i < n;i++) first[i][i] = nums[i];
for(int i = n-2;i >= 0;i--){
for(int j = i+1;j < n;j++){
second[i][j] = min(first[i+1][j],first[i][j-1]);
first[i][j] = max(nums[i]+second[i+1][j],nums[j]+second[i][j-1]);
}
}
return max(first[0][n-1],second[0][n-1]);
}
};
int main(){
int n;
cin >> n;
vector<int> nums(n);
for(int i = 0;i < n;i++) cin >> nums[i];
Solution *s = new Solution();
int res = s->getMaxScore(n,nums);
cout << res << endl;
}
486. 预测赢家
给定一个表示分数的非负整数数组。 玩家1从数组任意一端拿取一个分数,随后玩家2继续从剩余数组任意一端拿取分数,然后玩家1拿,……。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数总和最多的玩家获胜。
给定一个表示分数的数组,预测玩家1是否会成为赢家。你可以假设每个玩家的玩法都会使他的分数最大化。
示例 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可以成为赢家。
注意:
1 <= 给定的数组长度 <= 20.
数组里所有分数都为非负数且不会大于10000000。
如果最终两个玩家的分数相等,那么玩家1仍为赢家。
思路:
这里有个trick:
偶数个数字先手必胜直接返回true省时间,奇数个数字再按套路dp。
偶数个的话先手者一定可以拿到奇数序列或者偶数序列,哪个大拿哪个
class Solution {
public:
bool PredictTheWinner(vector<int>& nums) {
int n = nums.size();
if(n < 2) return true;
vector<vector<int>> first(n,vector<int>(n,0));
vector<vector<int>> second(n,vector<int>(n,0));
for(int i = 0;i < n;i++) first[i][i] = nums[i];
for(int i = n-2;i >= 0;i--){
for(int j = i+1;j < n;j++){
second[i][j] = min(first[i+1][j],first[i][j-1]);
first[i][j] = max(nums[i]+second[i+1][j],nums[j]+second[i][j-1]);
}
}
return first[0][n-1] >= second[0][n-1];
}
};