Acwing 110 畜栏预定 (贪心+stl)

题面

有N头牛在畜栏中吃草。

每个畜栏在同一时间段只能提供给一头牛吃草,所以可能会需要多个畜栏。

给定N头牛和每头牛开始吃草的时间A以及结束吃草的时间B,每头牛在[A,B]这一时间段内都会一直吃草。

当两头牛的吃草区间存在交集时(包括端点),这两头牛不能被安排在同一个畜栏吃草。

求需要的最小畜栏数目和每头牛对应的畜栏方案。

输入格式
第1行:输入一个整数N。

第2..N+1行:第i+1行输入第i头牛的开始吃草时间A以及结束吃草时间B,数之间用空格隔开。

输出格式
第1行:输入一个整数,代表所需最小畜栏数。

第2..N+1行:第i+1行输入第i头牛被安排到的畜栏编号,编号是从1开始的 连续 整数,只要方案合法即可。

数据范围
1≤N≤50000,
1≤A,B≤1000000
输入样例:
5
1 10
2 4
3 6
5 8
4 7
输出样例:
4
1
2
3
2
4

思路

一定要画图!!!!画图真的很重要,你把题目所示的区间画出来,然后逐个去添加栅栏,你会发现,你当前要不要新增栅栏只会和你当前所有的栅栏的最小右端点有关,这个点大于你的左端点的话意味着你没有办法加入已有栅栏,反之则可以。这里的一个贪心思想在于,我们有多个栅栏可以挑选的时候我们只需要考虑最小的就可以了,所以这个的话我们直接可以想到堆的做法,维护一个堆,去存储一个最小右端点和当前这个牛的所在栅栏号。对stl还是不大熟悉,需要加强训练。

代码实现

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<set>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=50010;
int n,m;
typedef pair <int,int  > PII;
pair <PII,int > cow[maxn];
int id[maxn];
int main () {
    cin>>n;
    for (int i=1;i<=n;i++) {
        cin>>cow[i].first.first>>cow[i].first.second;
        cow[i].second=i;
    }    
    sort (cow+1,cow+1+n);
    priority_queue <PII,vector<PII>,greater<PII> > q;
    for (int i=1;i<=n;i++) {
        PII cows=cow[i].first;
        if (q.empty ()||q.top().first>=cows.first) {
            PII tall={cows.second,q.size ()+1};
            id[cow[i].second]=tall.second;
            q.push (tall);
        }
        else  {
           PII tall=q.top ();
           q.pop ();
           tall.first=cows.second;
           id[cow[i].second]=tall.second;
           q.push (tall);
        }
    }
    cout<<q.size ()<<endl;
    for (int i=1;i<=n;i++) {
        cout<<id[i]<<endl;
    }
    return 0;
}
原文地址:https://www.cnblogs.com/hhlya/p/13326226.html