线段树--区间修改--cf1070-c Cloud Computing

n天每天需要k个东西，有m种价格方案：

方案i [Li,Ri] 内每天有ci个，每个pi元

（如果某天所能获得的东西不足k个，全部借完）

求最小花费。

解：

按价格排序，线段树记录某天已经借了的数量，成段更新。

更新时判断，1.如果最小值 >= k，即该区间已经满足条件，不用再借，return

2.如果最大值 + c <= k，即该区间全部都可以借c个，利用lazy标记区间更新

3.如果最大值==最小值，说明该区间所有值相等，可以一起借，借的个数为min(c,k - max)，区间更新

4.否则，向下pushdown

因此还需要l == r的递归出口

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
#define lc (rt << 1)
#define rc (rt << 1 | 1)
#define mid (l + r) / 2
const int maxn = 1e6 + 5;//

ll tr[maxn << 2];//记录第i天已经租的个数
int mn[maxn << 2],mx[maxn << 2];
int add[maxn << 2];
struct node{
    int l,r,c,p;
    bool operator < (const node &b) const{
        return p < b.p;
    }
}ns[maxn];

void init(int l,int r,int rt)
{
    if(l == r){
        tr[rt] = 0;
        mn[rt] = 0;
        mx[rt] = 0;
       
        return ;
    }
    init(l,mid,lc);
    init(mid + 1,r,rc);
    //pushup(l,r,rt);
    tr[rt] = tr[lc] + tr[rc];
    mn[rt] = min(mn[lc],mn[rc]);
    mx[rt] = max(mx[lc],mx[rc]);
}
int k;
ll ans = 0;
int c,p;
void pushdown(int l,int r,int rt)
{
    if(add[rt]){
        add[lc] += add[rt];add[rc] += add[rt];
        tr[lc] += 1ll * add[rt] * (mid - l + 1);
        tr[rc] += 1ll * add[rt] * (r - mid);
        mn[lc] += add[rt];mn[rc] += add[rt];
        mx[lc] += add[rt];mx[rc] += add[rt];
        add[rt] = 0;
    }
}
void update(int ql,int qr,int l,int r,int rt)
{

    if(l == r){//??
        int t = min(c,int(k - tr[rt]));
        ans += 1ll * t * p;
        tr[rt] = mx[rt] = mn[rt] = t + tr[rt];
        return;
    }
   
        if(ql <= l && r <= qr){
            if(mn[rt] >= k) return;
            if(mx[rt] + c <= k){
                add[rt] += c;
                tr[rt] += 1ll * (r - l + 1) * c;
                mn[rt] += c;mx[rt] += c;
                ans += 1ll * c * p * (r - l + 1);
                return;
            }
            else if(mx[rt] == mn[rt]){//mx[rt] + c > k
                int t = min(c,k - mn[rt]);
                add[rt] += t;
                tr[rt] += 1ll * (r - l + 1) * t;
                mn[rt] += t;mx[rt] += t;
                ans += 1ll * t * p * (r - l + 1);
                return;
            }
        }
    
    pushdown(l,r,rt);
    if(ql <= mid) update(ql,qr,l,mid,lc);
    if(mid < qr) update(ql,qr,mid + 1,r,rc);
    tr[rt] = tr[lc] + tr[rc];
    mn[rt] = min(mn[lc],mn[rc]);
     mx[rt] = max(mx[lc],mx[rc]);
}
int main()
{
    int n,m;
    scanf("%d%d%d",&n,&k,&m);
    for(int i = 0;i < m;i ++){
        scanf("%d%d%d%d",&ns[i].l,&ns[i].r,&ns[i].c,&ns[i].p);
    }
    sort(ns,ns + m);
    init(1, n, 1);
   
    for(int i = 0;i < m;i ++){
        c = ns[i].c;p = ns[i].p;
        update(ns[i].l,ns[i].r,1,n,1);
//        for (int i = 0; i < 2 * n; i ++) {
//            printf("%d %d %d %d %d\n",i,tr[i],mn[i],mx[i],add[i]);
//        }
//        printf("after %d th update ans = %lld \n",i,ans);
    }
    printf("%I64d\n",ans);
    return 0;
}