bzoj3638/3272 Cf172 k-Maximum Subsequence Sum 线段树模拟费用流

题意

给一列数，要求支持操作： 1.修改某个数的值 2.读入l,r,k，询问在[l,r]内选不相交的不超过k个子段，最大的和是多少。
1 ≤ n,m ≤ 105

分析

考虑费用流建模，就是每次找和最大的子段取反。用线段树模拟增广过程即可。
由于是第一次打，便去学习了一波黄学长的程序。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

const int N=100005;
const int inf=1000000000;

int n,m,a[N];
struct data{int ls,rs,l,r,pl,pr,s,mx;};
struct tree{data mn,mx;bool tag;}t[N*5];
vector<pair<int,int> > w;

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

data merge(data a,data b)
{
    data u;
    if (a.ls>a.s+b.ls) u.ls=a.ls,u.pl=a.pl;
    else u.ls=a.s+b.ls,u.pl=b.pl;
    if (b.rs>b.s+a.rs) u.rs=b.rs,u.pr=b.pr;
    else u.rs=b.s+a.rs,u.pr=a.pr;
    u.s=a.s+b.s;
    if (a.mx>b.mx) u.mx=a.mx,u.l=a.l,u.r=a.r;
    else u.mx=b.mx,u.l=b.l,u.r=b.r;
    if (a.rs+b.ls>u.mx) u.mx=a.rs+b.ls,u.l=a.pr,u.r=b.pl;
    return u;
}

void pushdown(int d,int l,int r)
{
    if (l==r||!t[d].tag) return;
    t[d].tag=0;
    t[d*2].tag^=1;swap(t[d*2].mn,t[d*2].mx);
    t[d*2+1].tag^=1;swap(t[d*2+1].mn,t[d*2+1].mx);
}

void build(int d,int l,int r)
{
    if (l==r)
    {
        t[d].mx.ls=t[d].mx.rs=t[d].mx.mx=t[d].mx.s=a[l];
        t[d].mx.l=t[d].mx.r=l;t[d].mx.pl=t[d].mx.pr=l;
        t[d].mn.ls=t[d].mn.rs=t[d].mn.mx=t[d].mn.s=-a[l];
        t[d].mn.l=t[d].mn.r=l;t[d].mn.pl=t[d].mn.pr=l;
        return;
    }
    int mid=(l+r)/2;
    build(d*2,l,mid);build(d*2+1,mid+1,r);
    t[d].mx=merge(t[d*2].mx,t[d*2+1].mx);
    t[d].mn=merge(t[d*2].mn,t[d*2+1].mn);
}

void modify(int d,int l,int r,int x,int y)
{
    pushdown(d,l,r);
    if (l==r)
    {
        t[d].mx.ls=t[d].mx.rs=t[d].mx.mx=t[d].mx.s=y;
        t[d].mn.ls=t[d].mn.rs=t[d].mn.mx=t[d].mn.s=-y;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) modify(d*2,l,mid,x,y);
    else modify(d*2+1,mid+1,r,x,y);
    t[d].mx=merge(t[d*2].mx,t[d*2+1].mx);
    t[d].mn=merge(t[d*2].mn,t[d*2+1].mn);
}

data query(int d,int l,int r,int x,int y)
{
    pushdown(d,l,r);
    if (l==x&&r==y) return t[d].mx;
    int mid=(l+r)/2;
    if (y<=mid) return query(d*2,l,mid,x,y);
    if (x>mid) return query(d*2+1,mid+1,r,x,y);
    return merge(query(d*2,l,mid,x,mid),query(d*2+1,mid+1,r,mid+1,y));
}

void rever(int d,int l,int r,int x,int y)
{
    if (x>y) return;
    pushdown(d,l,r);
    if (l==x&&r==y) {t[d].tag^=1;swap(t[d].mn,t[d].mx);return;}
    int mid=(l+r)/2;
    rever(d*2,l,mid,x,min(y,mid));
    rever(d*2+1,mid+1,r,max(x,mid+1),y);
    t[d].mx=merge(t[d*2].mx,t[d*2+1].mx);
    t[d].mn=merge(t[d*2].mn,t[d*2+1].mn);
}

void solve(int l,int r,int k)
{
    int ans=0;
    while (k--)
    {
        data u=query(1,1,n,l,r);
        if (u.mx<=0) break;
        rever(1,1,n,u.l,u.r);
        ans+=u.mx;w.push_back(make_pair(u.l,u.r));
    }
    for (int i=0;i<w.size();i++) rever(1,1,n,w[i].first,w[i].second);
    w.clear();
    printf("%d\n",ans);
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i]=read();
    build(1,1,n);
    m=read();
    while (m--)
    {
        int op=read();
        if (op==1)
        {
            int l=read(),r=read(),k=read();
            solve(l,r,k);
        }
        else
        {
            int x=read(),y=read();
            modify(1,1,n,x,y);
        }
    }
    return 0;
}