POJ(3190)(Stall Reservations)

题目链接:https://vjudge.net/problem/POJ-3190
意思:每个牛在某个区间工作,需要占用一个牛棚,问至少需要准备多少牛棚,并给出一个分配方案
思路:贪心加上优先级队列的题,需要注意,下一头牛开始时间必须是上一头牛结束时间+1才能用同一个牛棚。我采用的是用一个优先级队列(结束时间优先),一个数组排序(开始时间优先),然后一个普通队列(记录哪些牛棚编号可以用)。扫描每一头牛,首先将优先级队列中所有结束时间比当前牛开始时间早的弹出,并将对应牛棚编号压入普通队列(就是更新一遍对于该牛当前哪些牛棚可以用),紧接着先判断队列是否为空,如果为空则需要一个新牛棚,否则可以用空闲的原来有的牛棚,最后把当前牛的信息压入优先级队列,循环扫描所有牛的数据即可得到。
代码:
#include
#include
#include
#include
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 return a2.end }

bool cmp2(const cow &a1,const cow & a2){//数组排序,还原原来输入时的编号顺序
return a1.num }

int n;

int main(){
while(scanf("%d",&n)==1){
vector all;
cow now;
for(int i=0;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 nowr ;//记录空闲牛棚编号的队列
priority_queue judge;//以结束时间优先的优先级队列
all[0].qq = 1;
judge.push(all[0]);

    for(int i=1;ijudge.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 printf("%d\n",all[i].qq);
}
}
return 0;
}

你可能感兴趣的:(POJ(3190)(Stall Reservations))