【PAT A-1017】Queueing at Bank

【PAT A-1017】Queueing at Bank

C++代码

#include <bits/stdc++.h>
using namespace std;
using gg = long long;
using agg2 = array<gg, 2>;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    gg ni, ki;
    cin >> ni >> ki;
    gg startTime = 8 * 60 * 60, endTime = 17 * 60 * 60;
    priority_queue<gg, vector<gg>, greater<gg>> windows;
    priority_queue<agg2, vector<agg2>, greater<agg2>> customers;
    for (gg i = 0; i < ki; ++i) {
        windows.push(startTime);
    }
    while (ni--) {
        gg h, m, s, p;
        char c;
        cin >> h >> c >> m >> c >> s >> p;
        gg arriveTime = h * 60 * 60 + m * 60 + s;
        if (arriveTime > endTime) {
            continue;
        }
        customers.push({arriveTime, p * 60});
    }
    gg time = 0, num = customers.size();
    while (not customers.empty()) {
        auto c = customers.top();
        customers.pop();
        gg t = windows.top();
        windows.pop();
        if (t > c[0]) {
            time += t - c[0];
        }
        windows.push(max(c[0], t) + c[1]);
    }
    cout << fixed << setprecision(1) << time / 60.0 / num;
    return 0;
}