贪心算法牛挤奶----优先队列解法

有n头牛（1<=n<=50,000)要挤奶。给定每头牛挤奶的时间区间A,B。牛需要呆在畜栏里才能挤奶。一个畜栏同一时间只能容纳一头牛。问至少需要多少个畜栏，才能完成全部挤奶工作，以及每头牛都放哪个畜栏里？注意：在同一个畜栏的两头牛，它们挤奶时间区间不能在端点重合。

输入格式:
第1行：一个正整数N；
第2…N+1行：第i+1行的两个整数给出第i头奶牛的挤奶时间。

输出格式：
第1行：需要畜栏的最小数；
第2…N+1行：第i+1行表示第i头奶牛被分配到的畜栏序号

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

输出样例：
4
1
2
3
2
4

用优先队列存放已经分配的畜栏，并使得结束时间最早的畜栏始终 位于队列头部

#include 
#include 
#include 
#include 
using namespace std;
struct Cow {
    int b, e; // 挤奶区间起点和终点
    int No;   // 编号
    bool operator < (const Cow & c) const {
        return b < c.b;
    }
} cows[50100];

int pos[50100];   // pos[i]表示编号为i的奶牛去的畜栏编号
struct Stall {
    int end;   // 结束时间
    int No;    // 编号
    bool operator < (const Stall & s) const {
        return end > s.end;
    }
    Stall (int e, int n) : end(e), No(n) {}
};
int main() {
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%d%d", &cows[i].b, &cows[i].e);
        cows[i].No = i;
    }
    sort(cows, cows+n);
    int total = 0;
    priority_queue pq;
    for (int i = 0; i < n; i++) {
        if (pq.empty()) {
            ++total;
            pq.push(Stall(cows[i].e, total));
            pos[cows[i].No] = total;
        } else {
            Stall st = pq.top();
            if (st.end < cows[i].b) {   // 端点也不能重合
                pq.pop();
                pos[cows[i].No] = st.No;
                pq.push(Stall(cows[i].e, st.No));
            } else {
                ++total;
                pq.push(Stall(cows[i].e, total));
                pos[cows[i].No] = total;
            }
        }
    }
    printf("%d\n", total);
    for (int i = 0; i < n; i++) {
        printf("%d\n", pos[i]);
    }
    return 0;
}