【非原创】codeforces 1070C Cloud Computing 【线段树&树状数组】

题目：戳这里

学习博客：戳这里

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

解题思路：遍历每一天，维护当前天K个cpu的最小花费。具体方法是维护两个线段树（树状数组也可以），维护每一天可以使用的cpu数和价格*cpu数的前缀和。注意数组下标是价格（1e6的数组。

（不明白的话可以看代码，代码思路很清晰

附学习博客的代码：

#include <iostream>

#include <algorithm>

#include <string.h>

#include <vector>

#include <memory.h>

#include <bitset>

#include <map>

#include <deque>

#include <math.h>

#include <stdio.h>

using namespace std;

typedef long long int ll;

const int MAXN = 1000005;

 

ll num[MAXN<<2];

ll sum[MAXN<<2];

int N;

void pushup(int rt){

    num[rt]=num[rt<<1]+num[rt<<1|1];

    sum[rt]=sum[rt<<1]+sum[rt<<1|1];

}

 

void update(int P,int C,int l,int r,int rt){

    if(l==r){

        num[rt]+=C;

        sum[rt]+=1ll*P*C;

        return;

    }

 

    int m=(l+r)/2;

 

    if(P<=m)

        update(P,C,l,m,rt<<1);

    else

        update(P,C,m+1,r,rt<<1|1);

    pushup(rt);

}

 

ll query(int K,int l,int r,int rt){

 

    if(l==r){

        //不到K个

        if(l==MAXN){

            return 0;

        }

        if(K>0)

        {

            return 1ll*K*l;

        }

        else

            return 0;

    }

    int m=(l+r)/2;

    if(num[rt<<1]>=K){

        return query(K,l,m,rt<<1);

    }

    else{

        return sum[rt<<1]+query(K-num[rt<<1],m+1,r,rt<<1|1);

    }

}

 

vector<pair<int,int> > C[MAXN];//第i天加入的活动

vector<pair<int,int> > O[MAXN];//第i天结束的活动

 

int main()

{

    int K,M;

    scanf("%d%d%d",&N,&K,&M);

 

    int l,r,c,p;

    for(int i=0;i<M;i++){

        scanf("%d%d%d%d",&l,&r,&c,&p);

        C[l].push_back(make_pair(p,c));//加入的活动

        O[r].push_back(make_pair(p,c));//退出的活动

    }

 

    ll ans=0;

    for(int i=1;i<=N;i++){

        //新活动加入

        for(int j=0;j<C[i].size();j++)

            update(C[i][j].first,C[i][j].second,1,MAXN,1);

        ans+=query(K,1,MAXN,1);

        //活动结束

        for(int j=0;j<O[i].size();j++)

            update(O[i][j].first,-O[i][j].second,1,MAXN,1);

    }

    cout<<ans<<endl;

 

    return 0;

}