-
Notifications
You must be signed in to change notification settings - Fork 0
/
a58_m4_supply.cpp
39 lines (34 loc) · 930 Bytes
/
a58_m4_supply.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#include <bits/stdc++.h>
using namespace std;
int main() {
cin.tie(nullptr)->sync_with_stdio(false);
int N, M, K;
cin >> N >> M >> K;
vector <int> D(K), E(K), L(K);
for (int i = 0; i < K; i++) cin >> D[i] >> E[i] >> L[i];
vector <int> idx(K);
iota(idx.begin(), idx.end(), 0);
sort(idx.begin(), idx.end(), [&](const int &a, const int &b) {
return D[a] < D[b];
});
queue <int> plants, stores;
for (int i = 0; i < K; i++) {
int ans = 0;
if (E[idx[i]] == 0) {
if (!plants.empty()) {
ans = plants.front();
plants.pop();
}
else stores.push(L[idx[i]]);
}
else {
if (!stores.empty()) {
ans = stores.front();
stores.pop();
}
else plants.push(L[idx[i]]);
}
cout << ans << '\n';
}
return 0;
}