Contest1803 - 2019年第二阶段我要变强个人训练赛第十八场 问题 B: 无聊游戏 推公式+线段树

题目链接：http://icpc.upc.edu.cn/problem.php?cid=1803&pid=1

题目描述

小N和小A在玩这样的一个游戏：给定初始数列Q，小N先把某个前缀（可以为空）的数字全部乘上−A，小A再把某个后缀（可以为空）的数字全部乘上−B，小N想让最后所有数的和尽量大，而小A想让最后所有数的和尽量的小。
因为小A无比聪明绝对不会失误，所以小N想找到某个方法使得最后所有数的和尽量大，请帮助小N求出最大的值是多少吧。

 

输入

第一行三个正整数N,A,B，表示数列的长度和小N小A乘的数。
第二行有N个整数表示数列Q。

 

输出

输出一行一个整数S，表示最后数列的和的最大值。

 

样例输入

复制样例数据

3 1 1
-1 -2 -3

样例输出

0

 

提示

如果小N修改前0个数，那么小A修改后0个数，数列是{−1,−2,−3}，和为−6。
如果小N修改前1个数，那么小A修改后0个数，数列是{1,−2,−3}，和为−4。
如果小N修改前2个数，那么小A修改后0个数或后3个数，数列是{1,2,−3}或{−1,−2,3}，和为0。
如果小N修改前3个数，那么小A修改后3个数，数列是{1,−2,−3}，和为−6。
最后答案是max{−6,−4,0,−6}=0。
对于50%的数据，有n≤5000。
对于70%的数据，有n≤100000。
对于100%的数据，有n≤1000000,1≤A,B≤100,∣Qi∣≤109。

题解：第一个选前p，第二选q向后

p<q时，化简完：sum[p](A+1)-B*sum[n]+(B+1)*sum[q-1] 

p>=q时，-A(1+B)*sum[q-1]+B(A+B)sum[p]-B*sum[n]

所以枚举一下p，第一种，sum[q-1]，范围内取个最小值，第二种范围内取个最大值，线段树维护下即可，两种情况取个最小值，然后所有情况取最大值，还有注意第一个人不选的情况。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1000100;
struct node {
    int l, r;
    ll maxx, minn;
}tree[N << 2];
ll sum[N];
int n;
ll A, B;
void build(int l, int r, int cur) {
    tree[cur].l = l;
    tree[cur].r = r;
    if(l == r) {
        tree[cur].maxx = sum[l];
        tree[cur].minn = sum[l];
        return;
    }
    int mid=(r+l)>>1;
    build(l,mid,cur<<1);
    build(mid+1,r,cur<<1|1);
    tree[cur].maxx=max(tree[cur<<1].maxx,tree[cur<<1|1].maxx);
    tree[cur].minn=min(tree[cur<<1].minn,tree[cur<<1|1].minn);
}
ll query(int pl,int pr,int cur){
    if(pl<=tree[cur].l&&tree[cur].r<=pr){
        return tree[cur].maxx;
    }
    ll res=-1e18;
    if(pl<=tree[cur<<1].r)res=max(res,query(pl,pr,cur<<1));
    if(pr>=tree[cur<<1|1].l)res=max(res,query(pl,pr,cur<<1|1));
    return res;
}
ll query_(int pl,int pr,int cur){
    if(pl<=tree[cur].l&&tree[cur].r<=pr){
        return tree[cur].minn;
    }
    ll res=1e18;
    if(pl<=tree[cur<<1].r)res=min(res,query_(pl,pr,cur<<1));
    if(pr>=tree[cur<<1|1].l)res=min(res,query_(pl,pr,cur<<1|1));
    return res;
}
int main() {
    scanf("%d%lld%lld", &n,&A,&B); 
    ll x;
    for(int i = 1; i <= n; i++) {
        scanf("%lld", &x);
        sum[i] = sum[i - 1] +x;
    }
    build(1, n ,1);
    ll ans=-B*sum[n]+(B+1)*min(query_(1,n,1), 0LL) , cnt, tmp, t;
    for(int i=1;i<=n;i++){
    //  cout<<i<<endl;
        t = 1e18;
        if(i<=n){
            cnt=query_(i,n,1);
            tmp=-sum[i]*(A+1)-B*sum[n]+(B+1)*cnt;
        //  cout<<tmp<<endl;
            t=min(t,tmp);
        }
        if(i>1){
            cnt=query(1,i-1,1);
            cnt=max(cnt,0LL);
            tmp=-A*(B+1)*cnt+(B+A*B)*sum[i]-B*sum[n];
        //  cout<<tmp<<endl;
            t=min(t,tmp);
        }else{
            cnt=0;
            tmp=-A*(B+1)*cnt+(B+A*B)*sum[i]-B*sum[n];
        //  cout<<tmp<<endl;
            t=min(t,tmp);
        }
    //  cout<<t<<endl;
        ans=max(t,ans);
    }
    cout<<ans<<endl;
    return 0;
}