PAT 1017. Queueing at Bank (25)

#include<iostream>

#include<cstdio>

#include<string>

#include<cstring>

#include<queue>

#include<vector>

#include<cmath>

#include<iomanip>

#include<algorithm>

using namespace std;



const int MAX_N = 10000+5;

const int MAX_WIN = 100+5;

const int OFF_TIME = 9*60*60;

const int ONE_HOUR = 60*60;



struct Customer

{

	int iArrTime;  //到达的时间

	int iProTime;  //处理的时间

	int iWaitTime; //等待的时间

	bool isServed;//是否被处理

};

Customer arrCus[MAX_N];



struct Windows

{

	int iAvaiTime;//窗口available的时间

	

};

Windows arrWins[MAX_WIN];



//按到达时间进行排序

bool lessCmp(Customer cus1, Customer cus2)

{

	return cus1.iArrTime < cus2.iArrTime;

}



//将07:55:00形式的时间转换为相对08:00的时间，单位：秒。

//如果早于8点，则返回负数。

int getRelTime(char *chTime)

{

	int iRes;

	int h = (chTime[0] - '0')*10 + chTime[1]-'0';

	int m = (chTime[3] - '0')*10 + chTime[4]-'0';

	int s = (chTime[6] - '0')*10 + chTime[7]-'0';

	int iEight = 8*60*60;

	int iNow = (h*60 + m)*60 + s;

	iRes = iNow - iEight;

	return iRes;

}



void initData(int N,int K)

{

	int i,j;

	for(i=0; i<N; i++)

	{

		arrCus[i].iArrTime = 0;

		arrCus[i].iProTime = 0;

		arrCus[i].iWaitTime = 0;

		arrCus[i].isServed = true;

	}

	for(j=0; j<K; j++)

	{

		arrWins[j].iAvaiTime = 0;		

	}

}





//得到一个空闲的窗口,遍历K个窗口，找到窗口available最早的时间

int getAnFreeWin(int K)

{

	int iAvaTimeMin = arrWins[0].iAvaiTime;

	int iWinIndex = 0;

	for(int i=1; i<K; i++)

		if(arrWins[i].iAvaiTime < iAvaTimeMin)

		{

			iAvaTimeMin = arrWins[i].iAvaiTime;

			iWinIndex = i;

		}

	return iWinIndex;

}



int max(int a,int b)

{

	return a>b?a:b;

}



//窗口iWinId处理完一个客户后，需要更新窗口available最早的时间

void updateWinsInfo(int iWinId,int iCusId)

{

	int iCurTime = max(arrWins[iWinId].iAvaiTime,arrCus[iCusId].iArrTime);



	if(arrCus[iCusId].iProTime <= ONE_HOUR)//处理的时间不能超过一个小时

		arrWins[iWinId].iAvaiTime =iCurTime + arrCus[iCusId].iProTime;

	else

		arrWins[iWinId].iAvaiTime =iCurTime + ONE_HOUR;

}



//将iCusId客户放到窗口iWinId进行处理

void processAnCus(int iWinId, int iCusId)

{

	//如果在客户到之前就有窗口空闲，或者在下班以后窗口才空闲，则等待时间为0



	if(arrWins[iWinId].iAvaiTime == 0) //开始就空闲的窗口

	{

		if(arrCus[iCusId].iArrTime < 0)

			arrCus[iCusId].iWaitTime = abs(arrCus[iCusId].iArrTime);

		else if(arrCus[iCusId].iArrTime > OFF_TIME)

		{

			arrCus[iCusId].iWaitTime = 0;

			arrCus[iCusId].isServed = false;

		}

		else //没有等待，直接到窗口。

			arrCus[iCusId].iWaitTime = 0;

	}

	else//处理过客户的窗口

	{

		//if(arrWins[iWinId].iAvaiTime > OFF_TIME || arrCus[iCusId].iArrTime > OFF_TIME)//下班后的将不被处理

                if(arrCus[iCusId].iArrTime > OFF_TIME)//之前一直有个case过不了，发现是题目意思理解错了。。

                //即，客户只要在17:00之前到达银行，那么即使窗口的avaliable时间在17：00以后，那银行还是需要给

                //客户进行服务的

		{

			arrCus[iCusId].iWaitTime = 0;

			arrCus[iCusId].isServed = false;

		}

		else

		{

			if(arrWins[iWinId].iAvaiTime <= arrCus[iCusId].iArrTime)

				arrCus[iCusId].iWaitTime = 0;

			else

				arrCus[iCusId].iWaitTime = arrWins[iWinId].iAvaiTime - arrCus[iCusId].iArrTime;

		}

	}

		

	updateWinsInfo(iWinId,iCusId);

}



//获得所有客户的等待时间

double getAverageWaitTime(int N)

{

	int iSum = 0;

	int iCount = 0;

	for(int i=0; i<N; i++)

		if(arrCus[i].isServed)

		{

			iSum += arrCus[i].iWaitTime;

			iCount++;

		}

	if(iCount == 0)

		return 0;

	else

		return (double(iSum)/60)/iCount;

}



//对在等待队列上的客户进行处理

void process(int N, int K)

{

	sort(arrCus,arrCus+N,lessCmp);

	int iCusId = 0;

	while(iCusId < N)

	{

		int iFreeWinId = getAnFreeWin(K);

		processAnCus(iFreeWinId, iCusId);

		iCusId++;

	}

}



int main()

{	

	int N,K,i,j,iProTime;

	char chArrTime[10];

	scanf("%d %d",&N,&K);

	initData(N,K);

	for(i=0; i<N; i++)

	{

		scanf("%s",chArrTime);

		int iRelTime = getRelTime(chArrTime);

		arrCus[i].iArrTime = iRelTime;

		scanf("%d",&iProTime);

		arrCus[i].iProTime = iProTime*60;

	}

	process(N, K);

	cout<<fixed<<setprecision(1)<<getAverageWaitTime(N)<<endl;

	return 0;

}