【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;
}