PAT 1014 Waiting in Line (30)

#include <iostream>
#include <cstring>
#include <queue>
#define inf 10000
using namespace std;
typedef struct window{
	int endTime;
	queue<int> q;
}window;
typedef struct customer{
	int startTime;
	int endTime;
	int wasteTime;
}customer;
window win[22];//未初始化结构体成员变量都是0 
customer cus[1002];
int main(){
	int N,M,K,Q,i,j,minTime,location;//N个窗口，每个窗口最多M人排队，现在来了K个人员 
	int wasteTime[1002],queryNum[1002];
	cin>>N>>M>>K>>Q;
	for(i=1;i<=K;i++)
		cin>>cus[i].wasteTime;
	for(i=0;i<Q;i++)
		cin>>queryNum[i];
		/*下面尽量选用下标从1开始，因为涉及到顾客的编号，没有编号0，最小编号从1开始的*/
	for(i=1,j=1;i<=N*M&&i<=K;++i,j=i%N){//初始化一开始黄线以内等候排队服务的人,第i个客户等候在第j个窗口  
		cus[i].startTime=win[j].endTime;//一开始时候win[j].endTime都是0 
		cus[i].endTime=cus[i].startTime+cus[i].wasteTime;
		win[j].endTime=cus[i].wasteTime+win[j].endTime;
		win[j].q.push(i);//第i个人进入第j个窗口 
	}
	for(;i<=K;i++){//前面已经安排完毕N*M个人，接着安排
		minTime=inf;
		for(j=0;j<N;j++){
			int lastNum=win[j].q.front();//现在要从这N只队伍当中找出服务最先结束的那一只 	
			if(cus[lastNum].endTime<minTime)//找出每个队伍当中最后结束花费最短时间最短那个 
			{
				minTime=cus[lastNum].endTime;
				location=j;//把这个人安置在第j个窗口 
			}
		}
		cus[i].startTime=win[location].endTime;//又进来一个客户，重新更新上面数组 
		cus[i].endTime=cus[i].startTime+cus[i].wasteTime;
		win[location].endTime=cus[i].wasteTime+win[location].endTime;
		win[location].q.pop();//上面一个人服务结束 
		win[location].q.push(i);		 	
	}
	int hour,minute;
	for(i=0;i<Q;i++)
		if(cus[queryNum[i]].startTime+8*60>=17*60)
			cout<<"Sorry"<<endl;
		else{
			hour=cus[queryNum[i]].endTime/60+8;
			minute=cus[queryNum[i]].endTime%60;
			if(hour<10)
				cout<<"0";
			cout<<hour<<":";
			if(minute<10)
				cout<<"0";
			cout<<minute<<endl;
		} 
	return 0;
}