算法竞赛进阶指南——畜栏预订

畜栏预订

题目链接

#include<iostream>
using namespace std;
#include<algorithm>
#include<set>
#include<queue>

/*
输入样例：
5
1 10
2 4
3 6
5 8
4 7
输出样例：
4
1
2
3
2
4
*/


const int N=50005;

typedef pair<int,int> PII;

pair<PII,int> cows[N];
/*
cows[i].first.first     记录区间开始位置 
cows[i].first.second    记录区间结束位置
cows[i].second          记录区间下标 
*/

int id[N];              //记录每头牛用那个槽 

int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>cows[i].first.first>>cows[i].first.second;
		cows[i].second=i;
	} 
	sort(cows,cows+n);//默认按照first排序 
	priority_queue<PII,vector<PII>,greater<PII> > heap;
	//heap 中每个元素都是一个pair pair第一个元素记录该槽的结束时间  第二个元素记录改槽是第几个槽 
	for(int i=0;i<n;i++)
	{
		if(heap.empty() || heap.top().first >= cows[i].first.first)
		{
			PII stall={cows[i].first.second,heap.size()+1};
			id[cows[i].second]=stall.second;
			heap.push(stall);
		}
		else
		{
			auto stall=heap.top();
			heap.pop();
			id[cows[i].second]=stall.second;
			stall.first=cows[i].first.second;
			heap.push(stall);
		}
	}
	cout<<heap.size()<<endl;
	for(int i=0;i<n;i++) cout<<id[i]<<endl;
	return 0; 
}