题意
有n个石子,每个石子均有价值,Alice和Bob轮流操作,每次操作从当前剩余石子中选择前x个(x>1),然后把前x的石子价值和W累加到操作者,从当前剩余石子中去除前x个石子,添加一个价值W的石子在最左边。问在Alice和Bob想最大化两个人分数之差的情况下的结果是什么?
思路
观察发现,如果当前场上剩余k个石头,那么这k个石头一定是原序列的后k-1个石头+新石头(前n-k石头价值和)。在该观察的基础上,可以确定不管Alice、Bob如何操作,局面也只有n种情况。于是,考虑是否可以设dp[i]表示任意操作后,剩余石子的最左边编号为i的局面下,先手-后手的分数之差。若可以求出dp,那么答案就是dp[1]。(下标为1~n)
考虑边界值dp[n],因为先手无法操作,所以dp[n]=0。
考虑dp[i]如何更新?当前状态可以拿最左边2、3、...、n-i个,那么对应的值为sum[j]-dp[j],其中sum表示前缀和。因此dp[i]=max(sum[j]-dp[j]),其中j>i。
如果暴力更新dp[i],则复杂度为O(n^2)。观察dp[i]与dp[i-1]可以发现有公共部分max(sum[j]-dp[j]),其中j>i。dp[i-1]比dp[i]多出一项sum[i]-dp[i],因此可以贪心维护sum[i]-dp[i]最大值即可O(n)解决。
typedef long long ll;
class Solution {
public:
int n;
int stoneGameVIII(vector<int>& stones) {
n = stones.size();
vector<ll> dp(n + 1);
vector<ll> sum(n + 1, 0);
for(int i = 1; i <= n; i++){
sum[i] = sum[i - 1] + stones[i - 1];
}
dp[n] = 0; ll mx = -dp[n] + sum[n];
for(int i = n - 1; i >= 1; i--){
dp[i] = mx;
mx = max(mx, -dp[i] + sum[i]);
}
return dp[1];
}
};