维护一个堆,结束时间早的在堆顶,(堆中每个元素看做一台机器)把奶牛按照开始时间从小到大排序,对于每一头牛,如果当前堆顶的元素已结束,那么把该元素弹出,把当前牛加入(就是利用同一台机器),否则的话开一台新的机器,也就是把牛之间丢进堆里。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
struct machine {
int begin, end, num,rn;//还要记录牛原来的号码rn,否则排序后就乱了,num为机器编号
};
bool operator <(const machine&a, const machine&b) {
return a.end > b.end;
}
bool cmp(machine a, machine b)
{
return a.begin < b.begin;
}
priority_queue<machine>que;
int main()
{
int n, i, j, k;
machine cows[50005];
int cown[50005];
cin >> n; int ans = 0;
for (i = 1; i <= n; i++) {
scanf("%d %d", &cows[i].begin, &cows[i].end); cows[i].rn = i;
}
sort(cows + 1, cows + 1 + n, cmp);
for(i=1;i<=n;i++){
if (que.size() && que.top().end < cows[i].begin) {
int t = que.top().num; que.pop(); que.push(machine{ cows[i].begin,cows[i].end, t});
cown[cows[i].rn] = t;
}
else {
que.push(machine{ cows[i].begin,cows[i].end,++ans });
cown[cows[i].rn] = ans;
}
}
cout << ans << endl;
for (i = 1; i <= n; i++)
cout << cown[i] << endl;
return 0;
}