PAT 甲级 1017. Queueing at Bank (25)

题目：点击打开链接

思路：1.在输入时，可将达到时间晚于17:00:00的客户排除，不计入名单；

            2.先根据到达时间排序，将情况分为前K位和后K位（K为窗口数）达到两个部分：

               1）对前K位达到，若在8:00:00前达到，则等待时间则为其到8:00:00的时间，这里的服务时间是8:00:00，若在8:00:00后到达则无等待时间，这里的服务时间是其达到时间；

               2）对K位后，先将其之间的K为客户按照其离开时间排序（由于每次都对相应客户的前K位客户进行离开时间排序，最后整个向量排序都将根据离开时间排序，不必担心前面客户先到后出的情况发生），对排序后向前数第K位客户的离开时间作为标准，在此之前达到则有等待时间，否则等待时间为0；

          3.注意客户数小于窗口数的情况。

代码：

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

int final_time=17*60*60;
int begin_time=8*60*60;

struct customer
{
int arr_time;
int p;
int ser_time;
int leave_time;
};
//根据达到时间排序
bool comp_arr(customer &a,customer &b)
{
return a.arr_time<b.arr_time;
}

//根据离开时间排序
bool comp_leave(customer &a,customer &b)
{
return a.leave_time<b.leave_time;
}

//计算秒数
int get_time(string s)
{
int t=0;
t=(s[6]-'0')*10+(s[7]-'0');
t+=((s[3]-'0')*10+(s[4]-'0'))*60;
t+=((s[0]-'0')*10+(s[1]-'0'))*60*60;
return t;
}

int main()
{
int N,K;
cin>>N>>K;
int i=0;
vector<customer> C;
string s;
int t,q;
customer cum;

for(;i<N;++i)
{
cin>>s>>q;
t=get_time(s);
if(t<=final_time) //在17:00:01之前到达的客户
{
cum.arr_time=t;
cum.p=q;
C.push_back(cum);
}
}
sort(C.begin(),C.end(),comp_arr); //根据到达时间排序

long long w_time=0; //总的等待时间

if(C.size()<=K) //人数小于等于窗口数
{
for(i=0;i<C.size();++i)
{
if(C[i].arr_time<begin_time) //在开始时间之前来，有等待时间
{
w_time+=begin_time-C[i].arr_time;
C[i].ser_time=begin_time;
}
else
{
C[i].ser_time=C[i].arr_time;
}
C[i].leave_time=C[i].ser_time+C[i].p*60;
}
}
else
{
for(i=0;i<K;++i) //前K位客户，到8：00即可服务
{
if(C[i].arr_time<begin_time) //在开始时间之前来，有等待时间
{
w_time+=begin_time-C[i].arr_time;
C[i].ser_time=begin_time;
}
else
{
C[i].ser_time=C[i].arr_time;
}
C[i].leave_time=C[i].ser_time+C[i].p*60;
}
for(;i<C.size();++i)
{
sort(C.begin()+i-K,C.begin()+i,comp_leave); //对于前K位用户排序
if(C[i].arr_time<C[i-K].leave_time) //客户早到
{
w_time+=C[i-K].leave_time-C[i].arr_time;
C[i].ser_time=C[i-K].leave_time;
}
else
{
C[i].ser_time=C[i].arr_time;
}
C[i].leave_time=C[i].ser_time+C[i].p*60; //离开时间计算
}

}
float per_time=w_time/(60.0*C.size());
printf("%.1f",per_time);
system("pause");
return 0;
}