4144:畜栏保留问题

4144:畜栏保留问题

总时间限制: 

1000ms

内存限制: 

65536kB

描述

农场有N头牛，每头牛会在一个特定的时间区间[A, B]（包括A和B）在畜栏里挤奶，且一个畜栏里同时只能有一头牛在挤奶。现在农场主希望知道最少几个畜栏能满足上述要求，并要求给出每头牛被安排的方案。对于多种可行方案，主要输出一种即可。

输入

输入的第一行包含一个整数N(1 ≤ N ≤ 50, 000)，表示有N牛头；接下来N行每行包含两个数，分别表示这头牛的挤奶时间[Ai, Bi](1 ≤ A≤ B ≤ 1, 000, 000)。

输出

输出的第一行包含一个整数，表示最少需要的畜栏数；接下来N行，第i+1行描述了第i头牛所被分配的畜栏编号（从1开始）。

样例输入

5
1 10
2 4
3 6
5 8
4 7

样例输出

4
1
2
3
2
4

题目链接：

http://bailian.openjudge.cn/practice/4144

这个题和做任务三类似，但是多了一个标记。

做任务三详细介绍：

https://blog.csdn.net/qq_42217376/article/details/81156131

我直接来说做法二：按照左端点排序。给每头牛一个编号，如果不能做，那么这头牛就进新的畜栏。如果能做，那么这头牛就进结束最早的畜栏。优先队列维护结束时间。

AC代码：

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
struct A
{
    int s,e,mark;
    bool operator < (const A &a)const
    {
        return e>a.e;
    }
}a[50001];
priority_queue<A>q;
bool cmp(const A &a,const A &b)
{
    if(a.s==b.s)
        return a.e<b.e;
    return a.s<b.s;
}
int x[50001];//标记
int main()
{
    int n;
    scanf("%d",&n);
     for(int i=0;i<n;i++)
        {
            scanf("%d%d",&a[i].s,&a[i].e);
            a[i].mark=i;//先给每头牛一个编号
        }
        sort(a,a+n,cmp);
        int sum=0;
        for(int i=0;i<n;i++)
        {
            if(q.size()&&q.top().e<a[i].s)
            {
                x[a[i].mark]=x[q.top().mark];
                //能做，结束最早的来做。结束最早的那个人就是当前的那个人。
                q.pop();
                q.push(a[i]);
            }
            else
            {
                sum++;
                x[a[i].mark]=sum;//不能做，就有新加进来的人做
                q.push(a[i]);
            }
        }
        printf("%d\n",sum);
        for(int i=0;i<n;i++)
            printf("%d\n",x[i]);
    return 0;
}