牛客网小白月赛：希望（线段树+01背包）

Intention：给定一个长度为n的含负数的数组M, 要求在给定cost中，使其中的一些负数变为0，并求出数组和max

Strategy: 用线段树维护每个数字改变的最小代价，然后得到cost数组，value就是每个负数的绝对值（对答案的贡献），最后跑一边01背包；

Face

code:

#include<bits/stdc++.h>
// #include<bits/extc++.h>
#define oo INT_MAX
#define ll long long
#define db double
#define mp(a, b) make_pair(a, b)
#define met(a, b) memset(a, b, sizeof(a))
#define maxn 100009
#define _rep(i, a, b) for(int i = (a); i <= (b); ++i)
#define _rev(i, a, b) for(int i = (a); i >= (b); --i)
#define _for(i, a, b) for(int i = (a); i < (b) ;++i)
using namespace std;
int n, k, m;
ll sum;
struct node
{
    int v, l, r, p;
    #define le o << 1
    #define ri o << 1 | 1
    #define lson t[le]
    #define rson t[ri]
    
}t[maxn * 4];
int a[maxn], v[maxn];
int dp[maxn], w[maxn];
void pushup(int o){
    assert(o <= maxn * 4);
    t[o].v = min(lson.v, rson.v);
}

void pushdown(int o){
    assert(o <= maxn * 4);
    if(t[o].p){
        t[o].p = 0;
        if(t[o].v < lson.v){
            lson.v = t[o].v;
            if(lson.l != lson.r)lson.p = 1;
        }
        if(t[o].v < rson.v){
            rson.v = t[o].v;
            if(rson.l != rson.r)rson.p = 1;
        }
    }
}

void build(int o, int l, int r){
    assert(o <= maxn * 4);
    t[o].l = l, t[o].r = r;
    if(l == r){
        t[o].v = oo;
        return ;
    }
    int mid = (l + r) >> 1;
    build(le, l, mid);
    build(ri ,mid + 1, r);
    pushup(o);
}

int ask(int o,int l,int r){
    assert(o <= maxn * 4);
    if(t[o].l == l && t[o].r == r)
        return t[o].v;
    pushdown(o);
    int mid = (t[o].l + t[o].r) >> 1;
    if(mid >= r) return ask(le, l, r);
    else if(l > mid) return ask(ri, l, r);
    else return min(ask(le, l, mid), ask(ri, mid + 1, r));
}

void change(int o,int l, int r, int x){
    pushdown(o);
    if(l <= t[o].l && t[o].r <= r){
        if(t[o].v > x){
            t[o].v = x;
            if(t[o].l != t[o].r)t[o].p = 1;//标记代表，父节点已经修改，子节点尚未修改
        }
        return ;
    }
    int mid = (t[o].l + t[o].r) >> 1;
    if(mid >= l) change(le, l, r, x);
    if(r > mid)  change(ri, l, r, x);
	
}

int main(){
    ios::sync_with_stdio(0);
    cin >> n >> k >> m;
    build(1, 1, n);
    _rep(i, 1, n){
        cin >> a[i];
        sum += a[i];
    }
    _rep(i, 1, m){
        int l,r,x;
        cin >> l >> r >> x;
        change(1, l, r, x);
    }
    int cnt = 1;
    _rep(i, 1, n){
        if(a[i] < 0){
            v[cnt] = -a[i];
            w[cnt++] = ask(1, i, i);
        }
    }
    _rep(i, 1, cnt - 1){
        _rev(j, k, w[i]){
            dp[j] = max(dp[j], dp[j - w[i]] + v[i]);
        }
    }
    cout << sum + dp[k] << endl;
    system("pause");
}