懒得补了,每天坚持。想要写好代码,首先多写代码。
486. Predict the Winner
思路
一开始是误解了题目,以为拿到最大数字就赢了,然后发现完全不是那么回事。言归正传,虽然最后返回的是bool,dp过程的状态量仍然需要是一个数字。需要对题目的需求进行放大,得出一个包含结果的状态量。
参考了别人的解法,核心状态量为当前数组中先手可以比后手多得多少分,例如dp[i][j]
就是数组第i个元素到第j个元素作为范围,先手可以多得的分数。由于双方都想拿最高分,所以范围扩大的过程中只需要考虑第一个取的值减去剩下范围中对方多得的分数即可。
代码
class Solution {
public:
bool PredictTheWinner(vector<int>& nums) {
vector<vector<int>> dp(nums.size(), vector<int>(nums.size(), 0));
for (int i = 0; i < nums.size(); i++) {
dp[i][i] = nums[i];
}
for (int i = 1; i < nums.size(); i++) {
for (int j = 0; j < nums.size()-i; j++){
dp[j][j+i] = max(nums[j] - dp[j+1][j+i], nums[j+i] - dp[j][j+i-1]);
}
}
return dp[0].back() >= 0;
}
};