以石子问题为代表的最大最小问题是leetcode经常出现的一类博弈论类型的题目。主要的思路 贪心或者 DP。
877. 石子游戏
这里的dp设计是很有意思的一个设计,因为是最优博弈,因此双方都是希望与对方拉开最大差距的。所以我们可以直接定义dp[i][j]
为区间[i, j]中先手方可以拉开的最大分差。
这种定义的好处是没有了两人的区分,只是维护每个区间的先手最大优势就可以。
class Solution {
public boolean stoneGame(int[] piles) {
// 定义dp为区间内先手方可以领先的最大分差。
int n = piles.length;
int[][] dp = new int[n][n];
// 区间dp的标准模板,一个是倒叙一个是正序。并且需要考虑相等的情况
for (int i = n-1; i>=0; i--){
for(int j = i; j<n; j++){
if (i == j) dp[i][j] = piles[i];
else{
// 因为是选择某个石子的得分减去,在之后区间中可能落后的分数。(落后的分数也可能是负数)
dp[i][j] = Math.max(piles[i]-dp[i+1][j], piles[j]-dp[i][j-1]);
}
}
}
return dp[0][n-1]>0;
}
}
1140. 石子游戏 II
题目特点:拿走前N堆,每次拿走的堆数不同。返回的是先手方的得分。
基本思路还是dp的方法,这次的dp[i][j]
表示的是当前位置i
和M=j
的情况下,先手可以的最大得分。但是我们这里的枚举会相对麻烦一些。我们需要枚举位置和M,并且还要考虑再当前的M下,我们本次拿取应该拿取多少个。所以是三层的循环。
class Solution {
public int stoneGameII(int[] piles) {
int n = piles.length;
int[] lastsum = new int[n+1];
for (int i = n-1; i>=0;i--){
lastsum[i] = lastsum[i+1]+piles[i];
}
int[][] dp = new int[n+1][n+1];
for (int i = n-1; i>=0;i--){
// 枚举区间头位置
for(int j =