PAT 1014 Waiting in Line (30)

#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<cmath>
#include<iomanip>
using namespace std;

//本题容易出错的地方：是在17:00及以后开始服务的客户输出"Sorry"，而不是17:00之前结束服务的输出"Sorry"；
//如客户cId是16:59开始服务，服务时间为2，则输出的应该是17:01，而不是"Sorry"。

const int SIZE = 1005;

int T[1005];
int Query[1005];
int N,M,K,Q;

class Windows
{
public:    
	int curTime; //当前队首客户开始服务的时间。
	int curSerIndex;//等待队列的下标。
	int totalTime; //当前窗口队尾客户服务完成时所需要的时间。
	vector<int> vWait;
	Windows()
	{
		curTime = 0;
		curSerIndex = 0;
		totalTime = 0;
	}
};

struct Node
{
	int cId;//客户编号
	int serTime;//服务时间，以分钟为单位，从08:00开始算
};

//窗口信息数组
vector<Windows> vWin;
vector<Node> vCusTimeInfo;//保存每个客户的服务截止时间

int getAFreeWin()
{
	//获取队列人数最少的窗口，即最先完成服务的窗口；
	int tmpWinId = 0,i;
	//当前时间+在窗口等待的第一个人所需要的服务时间。
	int tmpIndex = vWin[0].curSerIndex;
	int minTime = vWin[0].curTime + T[vWin[0].vWait[tmpIndex]];
	for(i=1; i<N; i++)
	{
		tmpIndex = vWin[i].curSerIndex;
		int tmpCurTime = vWin[i].curTime + T[vWin[i].vWait[tmpIndex]];
		if(tmpCurTime < minTime)
		{
			tmpWinId = i;
			minTime = tmpCurTime;
		}
	}
	return tmpWinId;
}

//保存窗口为iWinId，客户编号为cId的客户服务所需时间信息。
void storeCusInfo(int iWinId, int cId)
{
		Node tmpNode;
		tmpNode.cId = cId;

		tmpNode.serTime = vWin[iWinId].totalTime;
		vCusTimeInfo.push_back(tmpNode);
}

//以正常的格式打印出时间信息
void printTimeInNormal(int iTime, int cId)
{
	char strHour[][3] = {"08","09","10","11","12","13","14","15","16","17"};
	int min = iTime%60;
	int hour = iTime/60;
	if( (iTime-T[cId]) >= 540) //是否能在17:00之前服务
		cout<<"Sorry"<<endl;
	else
	{
		if(min < 10)
			cout<<strHour[hour]<<":0"<<min<<endl;
		else
			cout<<strHour[hour]<<":"<<min<<endl;
	}
}


//找到用户id为iCusId的用户完成服务所需要的时间。
void printCusSerTime(int iCusId)
{
	vector<Node>::iterator it = vCusTimeInfo.begin();
	while(it != vCusTimeInfo.end())
	{
		if(it->cId == iCusId)//找到该用户的时间信息
		{
			printTimeInNormal(it->serTime,it->cId );
		}
		it++;
	}
}



int main()
{
	int i,j;

	//输入
	cin>>N>>M>>K>>Q;
	for(i=1; i<=K; i++)
		cin>>T[i];
	for(i=1; i<=Q; i++)
		cin>>Query[i];

	//初始化N个窗口
	for(i=0; i<N; i++)
	{	
		Windows *tmpWin = new Windows();
		vWin.push_back(*tmpWin);
	}
	
	//开始入队
	int curId = 1;

	for(i=0; i<M && curId <= K; i++) //每个窗口最多排M个人
	{
		for(j=0; j<N && curId <= K; j++)
		{
			vWin[j].vWait.push_back(curId);
			vWin[j].totalTime += T[curId];
			storeCusInfo(j,curId);
			curId++;
		}
	}

	//处理在等待队列上的客户
	while(curId <= K)
	{
	//	cout<<"vWin[0].totalTime: "<<vWin[0].totalTime<<endl;
	//	cout<<"vWin[1].totalTime: "<<vWin[1].totalTime<<endl;

		int iWinId = getAFreeWin();
		int tmpCurSerIndex = vWin[iWinId].curSerIndex;
		//完成了队首的服务，curSerIndex++
		vWin[iWinId].curSerIndex++;

		//tmpCid为在窗口iWinId等待的第一个客户
		int tmpCId = vWin[iWinId].vWait[tmpCurSerIndex];
		vWin[iWinId].curTime += T[tmpCId];
		
		//将当前的客户加入窗口iWinId中。
		vWin[iWinId].vWait.push_back(curId);
		vWin[iWinId].totalTime += T[curId];
		storeCusInfo(iWinId,curId);
		curId++;//处理在黄线外的等待队列的下一个客户
	}
//	cout<<"vWin[0].totalTime: "<<vWin[0].totalTime<<endl;

//	for(vector<Node>::iterator it=vCusTimeInfo.begin(); it!=vCusTimeInfo.end(); it++)
//	{
//		cout<<it->cId<<" "<<it->serTime<<endl;
//	}

	for(i=1; i<=Q; i++)
		printCusSerTime(Query[i]);
	return 0;
}