Codeforces 1070C - Cloud Computing 贪心set

题意:

有n天,每天需要用k个cpu,  然后给定m个计划,对于每个计划包含 L, R, c, p 表示,从第L天到第R天期间,每天你都可以选用c个cpu,每个cpu的花费为p; 问n天的最小花费;(当某天不能得到k个cpu时,就把能选的全选)

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
int n,k,m;

struct node
{
    int id,l,r;
    long long c,p;
    friend bool operator<(node a,node b)
    {
        if(a.p==b.p)
            return a.id in[1000005],out[1000005];

int main() {

    while(~scanf("%d%d%d",&n,&k,&m))
    {
        for(int i=0;i<=1000002;i++)
            in[i].clear(),out[i].clear();

        set s1,s2;
        for(int i=1;i<=m;i++)
        {
            long long l,r,c,p;
            scanf("%lld%lld%lld%lld",&l,&r,&c,&p);
            node tp(i,l,r,c,p);
            itm[i]=tp;
            in[l].push_back(i);
            out[r+1].push_back(i);
        }

        long long anss=0,sum=0,tans=0;
        for(int i=1;i<=n;i++)
        {
            int len1=out[i].size();
            for(int j=0;j::iterator it1,it2;
                it1=s1.end();it1--;
                it2=s2.begin();
                if(it1->p>it2->p)
                {
                    sum-=it1->c;tans-=it1->c*(it1->p);
                    sum+=it2->c;tans+=it2->c*(it2->p);

                    s1.insert(*it2);
                    s2.insert(*it1);
                    s1.erase(it1++);
                    s2.erase(it2++);
                }
                else
                    break;
            }

            if(sum>=k)
            {
                while(sum>=k)
                {
                    set::iterator it1=s1.end();
                    it1--;
                    if((sum-it1->c)c;
                        tans-=it1->c*it1->p;

                        s2.insert(*it1);
                        s1.erase(it1++);
                    }
                }
            }
            else
            {
                while(sum::iterator it1;
                    it1=s2.begin();
                    sum+=it1->c;
                    tans+=it1->c*it1->p;

                    s1.insert(*it1);
                    s2.erase(it1++);
                }
            }

            if(sum>k)
            {
                set::iterator it1;
                it1=s1.end();it1--;
                anss+=tans;
                anss-=it1->c*it1->p;
                long long tnum=k-(sum-it1->c);
                anss+=tnum*it1->p;
            } else
                anss+=tans;
        }
        printf("%lld\n",anss);
    }
    return 0;
}

 

你可能感兴趣的:(数据结构)