题目链接:传送门
思路:
按照题目思路
(1)如果人数少于n*m个人,不用等在黄线后面,直接找人少的队伍就行了,
所以设置一个数组tim记录m个窗口的最后一个人的结束时间,然后当n*m<k的情况下,依次从左向右入队,
然后更新tim数组。
(2)当黄线排满后,等前面的人处理完再处理,所以再次取出窗口中时间最短的一个,这里就需要使用优先队列了,
所以建立一个以结束时间先后为顺序的优先队列。
(3)查询时,访问每一个人的编号所在的入队时间判断入队时间是否在工作时间范围内即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 1200;
const int mxtim = (17-8)*60;
struct Node{
int pos,st_tim,pr_tim,ed_tim;
bool operator <(const Node tmp)const{
if(ed_tim==tmp.ed_tim) return pr_tim>tmp.pr_tim;
else return ed_tim>tmp.ed_tim;
}
}cur[maxn];
int last[maxn];
int main(void)
{
int n,m,k,Q,i,j;
scanf("%d%d%d%d",&n,&m,&k,&Q);
for(i=1;i<=k;i++) scanf("%d",&cur[i].pr_tim);
Node tp,tmp;
int cnt=0;
priority_queue <Node> q;
memset(last,0,sizeof(last));
for(i=1;i<=m&&cnt<k;i++){
for(j=0;j<n&&cnt<k;j++){
cnt++;
tmp.pos=j;
tmp.st_tim=last[j];
tmp.ed_tim=last[j]=tmp.st_tim+cur[cnt].pr_tim;
cur[cnt].pos=tmp.pos;
cur[cnt].st_tim=tmp.st_tim;
cur[cnt].ed_tim=tmp.ed_tim;
q.push(tmp);
}
}
for(i=cnt+1;i<=k;i++){
tmp=q.top();
q.pop();
tp.pos=tmp.pos;
tp.st_tim=last[tmp.pos];
tp.ed_tim=last[tmp.pos]=tp.st_tim+cur[i].pr_tim;
cur[i].pos=tp.pos;
cur[i].st_tim=tp.st_tim;
cur[i].ed_tim=tp.ed_tim;
q.push(tp);
}
int x,hh,mm;
for(i=1;i<=Q;i++){
scanf("%d",&x);
if(cur[x].st_tim>=mxtim) printf("Sorry\n");
else{
hh=cur[x].ed_tim/60+8;
mm=cur[x].ed_tim%60;
printf("%02d:%02d\n",hh,mm);
}
}
return 0;
}