#线段树,模拟费用流#CF280D k-Maximum Subsequence Sum

题目

给定一个大小为 \(n\) 的序列,要求支持单点修改和查询区间内至多 \(k\) 个不交子区间之和的最大值(可以不取)


分析

考虑源点向每个点、每个点向汇点流流量1费用0的边,每个点向右边的点流流量1费用\(a_i\)的边,流量最大为 \(k\),这样构建出一个费用流的模型。

很显然,退流相当于给区间取反,而可以利用反悔贪心,在线段树上维护最大子段和和最小子段和,在最大子段和大于0的情况下取反至多 \(k\) 次区间,查询完再恢复原状

复杂度 \(O(nk\log n)\)


代码

#include <cstdio>
#include <cctype>
#include <algorithm>
#include <stack>
using namespace std;
const int N=100011; int n,a[N];
stack<pair<int,int> >st;
int iut(){
    int ans=0,f=1; char c=getchar();
    while (!isdigit(c)) f=(c=='-')?-f:f,c=getchar();
    while (isdigit(c)) ans=ans*10+c-48,c=getchar();
    return ans*f;
}
void print(int ans){
    if (ans<0) putchar('-'),ans=-ans;
    if (ans>9) print(ans/10);
    putchar(ans%10+48);
}
struct Rec{int l,r,z;};
struct rec{
    Rec wmax,wmin,lmax,lmin,rmax,rmin; int sum,lazy;
    void ptag(int x,int y){
        wmax=wmin=lmax=lmin=rmax=rmin=(Rec){x,x,y};
        sum=y,lazy=0;
    }
    void rev(){
        swap(wmax,wmin),swap(lmax,lmin),swap(rmax,rmin);
        wmax.z=-wmax.z,wmin.z=-wmin.z,
        lmax.z=-lmax.z,lmin.z=-lmin.z,
        rmax.z=-rmax.z,rmin.z=-rmin.z;
        sum=-sum,lazy^=1;
    }
}w[N<<2];
rec pup(rec f,rec g){
    rec h; h.lazy=0;
    h.sum=f.sum+g.sum;
    if (f.lmax.z<f.sum+g.lmax.z) h.lmax=(Rec){f.lmax.l,g.lmax.r,f.sum+g.lmax.z};
        else h.lmax=f.lmax;
    if (f.lmin.z>f.sum+g.lmin.z) h.lmin=(Rec){f.lmin.l,g.lmin.r,f.sum+g.lmin.z};
        else h.lmin=f.lmin;
    if (g.rmax.z<f.rmax.z+g.sum) h.rmax=(Rec){f.rmax.l,g.rmax.r,f.rmax.z+g.sum};
        else h.rmax=g.rmax;
    if (g.rmin.z>f.rmin.z+g.sum) h.rmin=(Rec){f.rmin.l,g.rmin.r,f.rmin.z+g.sum};
        else h.rmin=g.rmin;
    h.wmax=f.wmax.z>g.wmax.z?f.wmax:g.wmax;
    if (h.wmax.z<f.rmax.z+g.lmax.z) h.wmax=(Rec){f.rmax.l,g.lmax.r,f.rmax.z+g.lmax.z};
    h.wmin=f.wmin.z<g.wmin.z?f.wmin:g.wmin;
    if (h.wmin.z>f.rmin.z+g.lmin.z) h.wmin=(Rec){f.rmin.l,g.lmin.r,f.rmin.z+g.lmin.z};
    return h;
}
void build(int k,int l,int r){
    if (l==r){
        w[k].ptag(l,a[l]);
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    w[k]=pup(w[k<<1],w[k<<1|1]);
}
void update(int k,int l,int r,int x,int y){
    if (l==r){
        w[k].ptag(x,y);
        return;
    }
    int mid=(l+r)>>1;
    if (w[k].lazy){
        w[k<<1].rev(),w[k<<1|1].rev();
        w[k].lazy=0;
    }
    if (x<=mid) update(k<<1,l,mid,x,y);
        else update(k<<1|1,mid+1,r,x,y);
    w[k]=pup(w[k<<1],w[k<<1|1]);
}
void flip(int k,int l,int r,int x,int y){
    if (l==x&&r==y){
        w[k].rev();
        return;
    }
    int mid=(l+r)>>1;
    if (w[k].lazy){
        w[k<<1].rev(),w[k<<1|1].rev();
        w[k].lazy=0;
    }
    if (y<=mid) flip(k<<1,l,mid,x,y);
    else if (x>mid) flip(k<<1|1,mid+1,r,x,y);
        else flip(k<<1,l,mid,x,mid),flip(k<<1|1,mid+1,r,mid+1,y);
    w[k]=pup(w[k<<1],w[k<<1|1]);
}
rec query(int k,int l,int r,int x,int y){
    if (l==x&&r==y) return w[k];
    int mid=(l+r)>>1;
    if (w[k].lazy){
        w[k<<1].rev(),w[k<<1|1].rev();
        w[k].lazy=0;
    }
    if (y<=mid) return query(k<<1,l,mid,x,y);
    else if (x>mid) return query(k<<1|1,mid+1,r,x,y);
        else return pup(query(k<<1,l,mid,x,mid),query(k<<1|1,mid+1,r,mid+1,y));
}
int main(){
    n=iut();
    for (int i=1;i<=n;++i) a[i]=iut();
    build(1,1,n);
    for (int Q=iut();Q;--Q)
    if (iut()){
        int l=iut(),r=iut(),k=iut(),ans=0;
        while (k--){
            rec t=query(1,1,n,l,r);
            if (t.wmax.z<=0) break;
            ans+=t.wmax.z;
            flip(1,1,n,t.wmax.l,t.wmax.r);
            st.push(make_pair(t.wmax.l,t.wmax.r));
        }
        print(ans),putchar(10);
        while (!st.empty()) flip(1,1,n,st.top().first,st.top().second),st.pop();
    }else{
        int x=iut(),y=iut();
        update(1,1,n,x,y);
    }
    return 0;
}
posted @ 2024-04-05 21:30  lemondinosaur  阅读(11)  评论(0编辑  收藏  举报