【ybtoj 高效进阶 1.2】 【贪心】 畜栏预定

【ybtoj 高效进阶 1.2】 【贪心】 畜栏预定

题目

---

解题思路

将奶牛们吃草的时间排序
然后查找是否有栅栏可以用
（就是上一头在这的奶牛是否吃完了）
最优的肯定是最早结束的
所以可以用小根堆维护最后一头牛吃完草的时间

---

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
struct lzf{
	int l,r,id;
}a[50010];
struct yty{
	int r,id;
	bool operator < (const yty a) const
	{
		return a.r<r;
	}
};
int n,ans[50020];
bool cmp(lzf l,lzf y)
{
	 return l.l<y.l;
}
int main()
{
	priority_queue<yty> q;
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
	    a[i].id=i;
	    scanf("%d%d",&a[i].l,&a[i].r);
	}
	sort(a+1,a+n+1,cmp);
	for (int i=1;i<=n;i++)
	    if (!q.size()||q.top().r>=a[i].l)
	    {
	       ans[a[i].id]=q.size()+1;  //新建一个栅栏
	       q.push((yty){a[i].r,ans[a[i].id]});   //放入小根堆
		}
		else {
			ans[a[i].id]=q.top().id;  //取这个栅栏的编号
			q.pop();  //上一头牛退堆
			q.push((yty){a[i].r,ans[a[i].id]});  //放入小根堆
		}
    printf("%d\n",q.size());
    for (int i=1;i<=n;i++)
        printf("%d\n",ans[i]);
    return 0;
}