202403 CSP-CCF 十滴水

#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
typedef long long int ll;
ll c, m, n;
set<ll>s;//这个是储存当前有水滴的格子的id，之所以存这个，是因为没用到没水滴的格子，这个和h搭配使用。
//这个玩意搭配迭代器就能省略while左右找的时间，最后只要返回大小即可
priority_queue<ll, vector<ll>, greater<ll>>q; //储存要判断和处理的格子id，只有一个参数
unordered_map<int, int>h; //储存格子id和里头的水滴个数

void func() {
	while (!q.empty()) {
		ll t = q.top();
		q.pop();
		if (h[t] >= 5) {
			h[t] = 0;
			set<ll>::iterator it = s.find(t);
			if (it != s.begin()) {
				it--;
				h[*it] += 1;
				q.push(*it);
			}
			it = s.find(t);
			if (++it != s.end()) {
				h[*it] += 1;
				q.push(*it);
			}
			s.erase(t);
		}
	}
}

int main(int argc, char **argv) {
	cin >> c >> m >> n; //分别表示有多少网格，有水的格子的个数以及操作次数
	ll pos, value, tar;

	while (m--) {
		cin >> pos >> value;
		h[pos] = value;
		s.insert(pos);
	}

	while (n--) {
		cin >> tar;
		h[tar] += 1;
		q.push(tar);
		func();
		cout << s.size() << endl;
	}
	return 0;
}