最小化價格(貪心,排序)

題目鏈接:最小化價格

鏈接:https://www.nowcoder.com/acm/contest/112/A
來源:牛客網

題目描述
現有n組人,m個地點,給出每組人的人數,每個地點可容納的最大人數和選擇的價格
要求一種方式,使得每組人都到一個各不相同的地點,最小化選擇的價格
每個隊伍的人都要在同一個地方每個地方只能有一個隊伍
輸入描述:
第一行n,m
第二行n個數,表示每組的人數
接下來m行,每行兩個數,表示可容納的最大人數和選擇的價格
輸出描述:
輸出最小化選擇的價格,無解輸出-1
示例1
輸入
3 4
2 3 4
1 2
2 3
3 4
4 5
輸出
12
備註:
所有數據小於1e5

思路:因爲每個隊伍的人都要在同一個地方,每個地方只能有一個隊伍,所以排序即可。

代碼:

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int maxn=1e5+9;

priority_queue<int,vector<int>,greater<int> >q;
int ar[maxn];
struct node
{
    int p,w;
    bool operator < (const node &a) const//從大到小排序
    {
        if(p==a.p) return w>a.w;
        return  p>a.p;
    }
}edge[maxn];

bool cmp(int a,int b)//從大到小
{
    return a>b;
}

int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0;i<n;i++) scanf("%d",&ar[i]);
        for(int i=0;i<m;i++) scanf("%d%d",&edge[i].p,&edge[i].w);
        sort(ar,ar+n,cmp);
        sort(edge,edge+m);


    for(int i=0;i<m;i++)
    {
        printf("%d %d\n",edge[i].p,edge[i].w);
    }


        int t=0;
        int flag=1,ans=0;
        while(!q.empty())q.pop();
        for(int i=0;i<n;i++)
        {
            while(ar[i]<=edge[t].p)
                q.push(edge[t].w),t++;
            if(q.empty())
            {
                flag=0;
                break;
            }
            ans+=q.top();
            q.pop();
        }
        if(flag)
            printf("%d\n",ans);
        else printf("-1\n");
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章