【贪心，小根堆优化】POJ 3190 Stall Reservations

链接

http://poj.org/problem?id=3190

大意

给定一些牛开始的休息时间和结束的时间，问最少要安排几间牛棚给它们休息，并输出方案

思路

先对牛开始休息的时间排序

维护一个数组$S$，保存每间牛棚休息结束时间最晚的那个牛棚，扫描$S$数组，放进能接上的所有牛棚结束时间最早的那个牛棚，若没有能够接上的牛棚，则新建一个牛棚
该算法的时间复杂度为$O(n^2)$

然后我们可以开一个小根堆，维护所有牛棚结束时间最早的那个牛棚，这样就可以在logn的时间内找到那头牛了
时间复杂度$O(nlogn)$

代码

#include<algorithm>
#include<cstdio>
#include<queue>
#define r(i,a,b) for(register int i=a;i<=b;i++)
using namespace std;int n,ans[50001],len=1;
struct node
{
    int st,ed,id;
    bool operator <(const node &a) const{
        return ed>a.ed||(ed==a.ed&&st>a.st);//大根堆变小根堆
    }
}cow[50001];
priority_queue<node>q;
bool cmp(node x,node y){return x.st<y.st;}
signed main()
{
    scanf("%d",&n);
    r(i,1,n) scanf("%d%d",&cow[i].st,&cow[i].ed),cow[i].id=i;
    sort(cow+1,cow+1+n,cmp);//排序
    q.push(cow[1]);ans[cow[1].id]=1;//初始化
    r(i,2,n)
    {
        if(q.size()&&q.top().ed<cow[i].st)
        {
            ans[cow[i].id]=ans[q.top().id];//新来的牛用原本的牛的牛棚
            q.pop();//该牛休息完毕，退出
        }
        else ans[cow[i].id]=++len;//无法接上，新开一个牛棚
        q.push(cow[i]);//又进入了一头牛
    }
    printf("%d\n",len);//输出
    r(i,1,n) printf("%d\n",ans[i]);//输出
}