Codeforces1420 C2.Pokémon Army (hard version)(思维+差分)

题意:

在这里插入图片描述

数据范围:n,q<=3e5

解法:
考虑计算出原数组的差分数组,
因为-a[i]+a[j],相当于选择差分数组中的区间[i+1,j],
不过题目是+a[i]-a[j],那么逆向求差分数组,变为选择[i,j-1].
那么问题变为在差分数组上找出若干个不相交区间,满足和最大.
此外,差分数组上选[1,2]其实等同于选[1,1][2,2],
因此其实就是在差分数组上选择若干个数,满足和最大,
那么贪心地选择大于0的即可.
交换两个数之后,只需要减去旧贡献,更新差分数组,
然后再加上新贡献即可.
code:
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=3e5+5;
int a[maxm];
int b[maxm];
int n,q;
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    int T;cin>>T;
    while(T--){
        cin>>n>>q;
        for(int i=1;i<=n;i++){
            cin>>a[i];
        }
        for(int i=1;i<=n;i++){
            b[i]=a[i]-a[i-1];
        }
        int ans=0;
        for(int i=1;i<=n;i++){
            if(b[i]>0)ans+=b[i];
        }
        cout<<ans<<endl;
        for(int i=1;i<=n;i++){
            ;
        }
        while(q--){
            int l,r;cin>>l>>r;
            //
            if(b[l]>0)ans-=b[l];
            if(b[r]>0)ans-=b[r];
            if(l!=n&&b[l+1]>0)ans-=b[l+1];
            if(r!=n&&b[r+1]>0)ans-=b[r+1];
            if(l+1==r&&b[r]>0)ans+=b[r];//多减了一次,加回来
            //
            swap(a[l],a[r]);
            b[l]=a[l]-a[l-1];
            b[r]=a[r]-a[r-1];
            if(l!=n)b[l+1]=a[l+1]-a[l];
            if(r!=n)b[r+1]=a[r+1]-a[r];
            //
            if(b[l]>0)ans+=b[l];
            if(b[r]>0)ans+=b[r];
            if(l!=n&&b[l+1]>0)ans+=b[l+1];
            if(r!=n&&b[r+1]>0)ans+=b[r+1];
            if(l+1==r&&b[r]>0)ans-=b[r];//多加了一次,减回去
            //
            cout<<ans<<endl;
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值