leetcode 1872. 石子游戏 VIII

 题意

有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];
    }
};