题目链接:https://vjudge.net/problem/POJ-3190
意思:每个牛在某个区间工作,需要占用一个牛棚,问至少需要准备多少牛棚,并给出一个分配方案
思路:贪心加上优先级队列的题,需要注意,下一头牛开始时间必须是上一头牛结束时间+1才能用同一个牛棚。我采用的是用一个优先级队列(结束时间优先),一个数组排序(开始时间优先),然后一个普通队列(记录哪些牛棚编号可以用)。扫描每一头牛,首先将优先级队列中所有结束时间比当前牛开始时间早的弹出,并将对应牛棚编号压入普通队列(就是更新一遍对于该牛当前哪些牛棚可以用),紧接着先判断队列是否为空,如果为空则需要一个新牛棚,否则可以用空闲的原来有的牛棚,最后把当前牛的信息压入优先级队列,循环扫描所有牛的数据即可得到。
代码:
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
struct cow{
int start;//开始时间
int end;//结束时间
int num;//给牛的编号
int qq;//牛所用牛棚的编号
bool operator < (const cow & a)const{//优先级队列,结束时间小的优先
return end > a.end;
}
};
bool cmp(const cow &a1,const cow & a2){//数组排序,开始时间小的优先
if(a1.start!=a2.start)
return a1.start<a2.start;
return a2.end<a2.end;
}
bool cmp2(const cow &a1,const cow & a2){//数组排序,还原原来输入时的编号顺序
return a1.num<a2.num;
}
int n;
int main(){
while(scanf("%d",&n)==1){
vector<cow> all;
cow now;
for(int i=0;i<n;i++){
scanf("%d%d",&now.start,&now.end);
now.num = i+1;
all.push_back(now);
}
sort(all.begin(),all.end(),cmp);//按照开始时间大小排序
int result = 1;
queue<int> nowr ;//记录空闲牛棚编号的队列
priority_queue<cow> judge;//以结束时间优先的优先级队列
all[0].qq = 1;
judge.push(all[0]);
for(int i=1;i<n;i++){
while(all[i].start>judge.top().end){//更新一遍当前可用牛棚的状态
nowr.push(judge.top().qq);
judge.pop();
}
if(nowr.size()==0){//先判断是否为空,为空则增加最小牛棚数量
result++;
all[i].qq = result;
}
if(nowr.size()!=0){//不为空则用已有的空闲牛棚
all[i].qq = nowr.front();
nowr.pop();
}
judge.push(all[i]);//将当前牛状态压入优先级队列并更新
}
printf("%d\n",result);
sort(all.begin(),all.end(),cmp2);
for(int i = 0;i<n;i++){
printf("%d\n",all[i].qq);
}
}
return 0;
}