权值线段树

https://blog.csdn.net/lzc504603913/article/details/83240651

题意：有很多个活动，每个活动有持续天数，每个活动会在每天提供C个CPU每个CPU价格为P，问需要工作N天，每天需要K个CPU的最少花费。

解题思路：首先价格越低的活动，肯定是要选的。因此我们对于每一天记录有哪些新活动 加入，哪些活动结束。然后维护一个线段树，线段树的下标是价格。即价格为i的活动，一共能提供多少个CPU，然后加入和删除活动就相当于update(C,+-P,1,MAXC,1)。 然后我们顺便维护一下价格*数量的和。然后利用线段树天然的二分性，快速求出前缀数量和为K的价格和。

/*************************************************************************
    > File Name: a.cpp
    > Author: QWX
    > Mail: 
    > Created Time: 2018/11/19 22:39:42
 ************************************************************************/


//{{{ #include
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<string>
#include<cstring>
#include<complex>
#include<cassert>
//#include<bits/stdc++.h>
#define vi vector<int>
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define pw(x) (1ll << (x))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define rep(i,l,r) for(int i=(l);i<(r);i++)
#define per(i,r,l) for(int i=(r);i>=(l);i--)
#define FOR(i,l,r) for(int i=(l);i<=(r);i++)
#define cl(a,b) memset(a,b,sizeof(a))
#define fastio ios::sync_with_stdio(false);cin.tie(0);
#define lson l , mid , ls
#define rson mid + 1 , r , rs
#define INF 0x3f3f3f3f
#define LINF 0x3f3f3f3f3f3f3f3f
#define ll long long
#define ull unsigned long long
#define dd(x) cout << #x << " = " << (x) << "," 
#define de(x) cout << #x << " = " << (x) << "
" 
#define endl "
"
using namespace std;
//}}}
const int N=1e6+7;

struct Set{
#define ls rt<<1
#define rs ls|1
    static const int N=::N<<2;
    ll sum[N],cnt[N];
    void up(int rt){
        cnt[rt]=cnt[ls]+cnt[rs];
        sum[rt]=sum[ls]+sum[rs];
    }
    void upd(int p,ll c,int l,int r,int rt){
        if(l==r){
            cnt[rt]+=c;
            sum[rt]=cnt[rt]*l;
            return ;
        }
        int mid=l+r>>1;
        if(p<=mid)upd(p,c,l,mid,ls);
        else upd(p,c,mid+1,r,rs);
        up(rt);
    }
    ll qry(ll k,int l,int r,int rt){
        if(cnt[rt]<=k)return sum[rt];
        int mid=l+r>>1; 
        if(l==r)return k*l;
        if(cnt[ls]>=k)return qry(k,l,mid,ls);
        return sum[ls]+qry(k-cnt[ls],mid+1,r,rs);
    }
}seg;
vector<pii>V[N];

int main()
{
    int n,k,m; cin>>n>>k>>m;
    ll ans=0;
    rep(i,0,m){
        int a,b,c,d; cin>>a>>b>>c>>d;
        V[a].pb(mp(c,d));V[b+1].pb(mp(-c,d));
    }
    FOR(i,1,n){
        for(auto &t:V[i]){
            seg.upd(t.se,t.fi,1,1e6,1);
        }
        ans+=seg.qry(k,1,1e6,1);
    }
    cout<<ans<<endl;
    return 0;
}