CodeForces - 371D Vessels 【并查集】

Vessels

---

题意： 自上而下的n个碗，向某个碗中倒水，如果溢出，则会流向它之下的下一个未满的碗。有两种操作：1 p x表示往第p个碗中导入x的水，2 p表示询问此时第p个碗中的水量。

题解： 未经思考地暴力走了一发，T在了前八组数据。然后再想，某些连续的的溢出的碗，可以选择合并处理，这个合并，就用到了并查集。

_{我没想到的是我提交了G++ 17 9.2.0 以后居然RE了，换了老一点的版本又能通过。}

#include<bits/stdc++.h>
using namespace std;

const int N = 200010;
int s[N], w[N], c[N];
int find(int x) { return s[x] == x ? x : s[x] = find(s[x]);}
int un(int a, int b) {
	a = find(a); b = find(b);
	s[a] = b;
}

int main() {
	ios :: sync_with_stdio(false); cin.tie(0), cout.tie(0);
	int n; cin >> n;
	for(int i = 1; i <= n; i++)	 	cin >> w[i];
		for(int i = 1; i < N; i++)	s[i] = i, c[i] = 0;
		int q; cin >> q;
		while(q--) {
			int k; cin >> k;
			if(k == 1) {
				int p, x;
				cin >> p >> x;
				while(x > 0) {
					p = find(p);
					if(p > n) break;
					if(c[p] + x >= w[p]) {
						x -= (w[p] - c[p]);
						c[p] = w[p];
						un(p, p + 1);
					} else {
						c[p] += x;
						x = 0;
					}
				}
			} else {
				int p; cin >> p;
				cout << c[p] << endl;
		}	
	}
	return 0;
}