PAT甲级1017

PAT甲级1017

题目大意：给出总人数n，银行窗口k，接下来k行给出每个人来的时间以及办理业务所需的时间，求出平均等待时间（银行8点开门17点关门，可以早来但是晚于17点的不计算在内）。
简单的模拟，储存每个人来的时间以及需要占用窗口的时间，实时更新可用的窗口数量，如果小于k个则不用等待直接用；如果等于k个则需要等到最早用完窗口的人离开，为了方便加入删除给vector倒序排列，每次窗口满弹出最后一个结束时间即可。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int n,k,sum_p=0,sum_t=0;
double average=0;
struct person{
	int start,process;
};
bool cmp(person a,person b){return a.start<b.start;}
bool cmp2(int a,int b){return a>b;}
int timeToInt(int hour,int minute,int second){
	return hour*3600+minute*60+second;
}
void update(vector<int> &windows,int start){
	for(int i=windows.size()-1;i>=0;i--){
		if(start>=windows[i]){
			windows.pop_back();
		}
		else break;
	}
}
int main(){
	scanf("%d%d",&n,&k);
	vector<person> line;
	vector<int> windows;
	int hour,minute,second,p,open=timeToInt(8,0,0),close=timeToInt(17,0,0);
	for(int i=0;i<n;i++){
		scanf("%d:%d:%d %d",&hour,&minute,&second,&p);
		int tmpstart=timeToInt(hour,minute,second);
		if(tmpstart>close) continue;
		else{
			person tmpperson;
			tmpperson.start=tmpstart,tmpperson.process=timeToInt(0,p,0);
			line.push_back(tmpperson);
		}
	}
	sort(line.begin(),line.end(),cmp);
	for(int i=0;i<line.size();i++){
		int wait,firstend;
		sum_p++;
		update(windows,line[i].start);
		if(line[i].start<open){
			wait=open-line[i].start;
			sum_t+=wait;
			line[i].start=open;
		}
		if(windows.size()>=k){
			firstend=windows[k-1];
			wait=firstend-line[i].start;
			sum_t+=wait;
			windows.pop_back();
			windows.push_back(firstend+line[i].process);
		}else{
			windows.push_back(line[i].start+line[i].process);
		}
		sort(windows.begin(),windows.end(),cmp2);
	}
	printf("%.1f",(sum_t*1.0/60)/sum_p);
	system("pause");
	return 0;
}