acwing111-畜栏预定

算法分类：

区间分组的应用 贪心

---

问题描述

有 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

---

实现步骤：

将所有牛按开始吃草的时间排序；
用小根堆维护当前所有畜栏的最后一头牛的吃草结束时间；
如果当前的牛可以安排在堆顶的畜栏中，则将其安排进去，否则就新建一个畜栏。

经典的区间分组问题。

---

实现代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define x first
#define y second
using namespace std;
const int N = 1e5+10;
typedef pair<int,int> PII;
//cows.x.x为当前牛的吃草开始时间 cows.x.y为当前牛的吃草结束时间 cows.y为牛的编号
pair<PII,int> cows[N];
int n;
int id[N];
int main(){
    cin >> n;
    for(int i=0;i<n;i++){
        int l,r;
        cin>>l>>r;
        cows[i].x.x = l;
        cows[i].x.y = r;
        cows[i].y = i;//存储编号
    }
    sort(cows,cows+n);

    priority_queue<PII,vector<PII>, greater<PII> > heap;
    for(int i=0;i<n;i++){
        if(heap.empty()||heap.top().x>=cows[i].x.x){
            id[cows[i].y] = heap.size();
            heap.push({cows[i].x.y,heap.size()});
        }
        else{
            auto stall = heap.top();
            heap.pop();
            stall.x = cows[i].x.y;
            heap.push(stall);
            id[cows[i].y] = stall.y;
        }
    }
    cout<<heap.size()<<endl;
    for(int i=0;i<n;i++) cout<<id[i]+1<<endl;
    return 0;
}