[补题记录] Atcoder Beginner Contest 320（E）

URL：https://atcoder.jp/contests/abc320

目录

E

Problem/题意

Thought/思路

Code/代码

---

E

Problem/题意

有 N 个人和 M 个事件，每个人有自己固定的编号。

每个事件在 t 秒时会产生 w 的面条，此时编号最小的人获得该面条，但是也需要在 s 秒后才能回到原来的位置。

问 M 个事件后，每个人获得面条的情况。

Thought/思路

用一个 set<int> st1 保存队伍中的人，用一个 set<pair<int, int>> st2 保存暂时被移除的人以及他回归的时间（时间作为 first）。

那么，当一个时间发生时，先判断 st2 内是否有可以放回 st1 的人，然后判断 st1 是否还有人。如果 st1 还有人，则取出第一个，令其获得面条，然后移除即可。

Code/代码

#include "bits/stdc++.h"

#define int long long

int n, m, ans[200005];

signed main() {
	std::cin >> n >> m;

	std::set <int> st;
	for (int i = 1; i <= n; ++ i) st.insert(i);

	std::set <std::pair <int, int>> timeToBack;
	for (int i = 1; i <= m; ++ i) {
		int t, w, s; std::cin >> t >> w >> s;

		while (timeToBack.size()) {
			auto &pair = *timeToBack.begin();
			if (pair.first <= t) {
				st.insert(pair.second);
				timeToBack.erase(timeToBack.begin());
			} else {
				break;
			}
		}

		if (st.size() == 0) continue;

		ans[*st.begin()] += w;
		timeToBack.insert({t + s, *st.begin()});
		st.erase(st.begin());

	}

	for (int i = 1; i <= n; ++ i) std::cout << ans[i] << "\n";
}