原题链接:Minimum Average Waiting Time
题目大意:顾客在不同的时刻到达餐馆,点的pizza不同,需要花的时间也不同,目标是让顾客的平均等待时间最小。注意:厨师一旦开始做一个pizza,只有这个pizza做完后才能开始下一个,而且厨师不知道还没来的顾客会点什么pizza。
大致思路:这是一个很典型的任务调度问题,和CPU的进程调度非常相似。由于厨师一次只能做一个pizza,且一定要完成后才能开始下一个,所以可以将厨师看成非抢占式的CPU。对于非抢占式的CPU,最优的调度算法是SJF(Shortest Job First,最短作业优先),这里就不证明了。由于在一个顾客来之前,我们无法预知他点的pizza要花多少时间,所以不能简单地将所有顾客按任务所需时间从小到大排序;我们应该将输入数据想像成是按时间顺序到达的,在某个顾客服务完成之后,将当前已经点餐的顾客按任务所需时间从小到大排序,服务排在最前面的顾客,并在结束服务后重复该过程。具体实现如下:
#include <cstdio>
#include <algorithm>
using namespace std;
struct cus {
long long t;
long long l;
bool operator < (const cus& p) const {
return t < p.t;
}
};
cus arr[100005];
bool cmp (const cus& p1, const cus& p2) {
return p1.l < p2.l;
}
int main() {
int T;
scanf("%d", &T);
for(int i=0; i<T; i++) {
scanf("%lld%lld", &arr[i].t, &arr[i].l);
}
sort(arr, arr+T);
long long endtime = arr[0].t + arr[0].l;
arr[0].l = endtime;
int j = 0;
for(int i=1; i<T; i++) {
j = i;
while(j<T && arr[j].t <= endtime) {
j++;
}
sort(arr+i, arr+j, cmp);
if(endtime >= arr[i].t) arr[i].l += endtime;
else arr[i].l += arr[i].t;
endtime = arr[i].l;
}
long long ans = 0;
for(int i=0; i<T; i++)
ans += arr[i].l - arr[i].t;
printf("%lld\n", ans/T);
return 0;
}
上面的实现中,在服务一位顾客后,我们为当前已经点餐的顾客进行了排序,这一步可以改用数据结构<set>实现,具体代码如下:
#include <cstdio>
#include <set>
#include <algorithm>
using namespace std;
struct cus {
long long t;
long long l;
bool operator < (const cus& p) const {
return t < p.t;
}
};
cus arr[100005];
int main() {
int T;
scanf("%d", &T);
for(int i=0; i<T; i++) {
scanf("%lld%lld", &arr[i].t, &arr[i].l);
}
sort(arr, arr+T);
long long ans = 0, endtime = 0;
multiset<cus> _list;
for(int i=0; i<T || _list.size()>0;) {
if(_list.size()==0)
endtime = max(arr[i].t, endtime);
while(i<T && arr[i].t <= endtime) {
_list.insert((cus){arr[i].l, arr[i].t});
i++;
}
endtime += _list.begin()->t;
ans += endtime - _list.begin()->l;
_list.erase(_list.begin());
}
printf("%lld\n", ans/T);
return 0;
}
注意,第30行将当前已点餐的顾客加入队列时,需将任务所需时间放在前面,这样 set 就会按照任务所需时间的大小排好序,接下来只要每次取队首的元素即可。