【YbtOJ高效进阶 贪心-3】畜栏预定

链接

YbtOJ高效进阶 贪心-3

题目描述

有N头牛在畜栏中吃草。每个畜栏在同一时间段只能提供给一头牛吃草，所以可能会需要多个畜栏，给出第i头牛开始吃草的时间区间$[a_i,b_i]$ ，求需要的最少畜栏数和每头牛对应的畜栏方案。

样例输入

5
1 10
2 4
3 6
5 8
4 7

样例输出

4
1
2
3
2
4

思路

对每头牛的开始时间从小到大排序
然后贪心？
对于第i个牛，它要么是进新的畜栏，要么是进已经没牛的畜栏
1.进新的畜栏，那只有每个畜栏都有牛，才会开新的畜栏给这头牛
2.进没牛的畜栏，那就要对每个畜栏中的牛的区间用一个小根堆去维护，然后判断即可

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>

using namespace std;

int n, ans[100005];
priority_queue <pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > Q;

struct tt 
{
    
    
	int l, r, num;
}a[100005];


bool cmp(tt a, tt b)
{
    
    
	return a.l < b.l;
}

int main()
{
    
    
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i)
		scanf("%d%d", &a[i].l, &a[i].r), a[i].num = i;
	sort(a + 1, a + n + 1, cmp);
	Q.push(make_pair(a[1].r, 1));
	ans[a[1].num] = 1;
	for(int i = 2; i <= n; ++i)
	{
    
    
		if(!Q.size() || Q.top().first >= a[i].l)	
		{
    
    
			ans[a[i].num] = Q.size() + 1;
			Q.push(make_pair(a[i].r, Q.size() + 1));
		}//对应思路中的第一种情况
		else {
    
    
			ans[a[i].num] = Q.top().second; Q.pop();
			Q.push(make_pair(a[i].r, ans[a[i].num]));
		}//第二种情况
	} 
	printf("%d\n", Q.size());
	for(int i = 1; i <= n; ++i)
		printf("%d\n", ans[i]);
}