[ABC366C] Balls and Bag Query 题解

[ABC366C] Balls and Bag Query

维护一个 $sum$ 值，为当前数字的种类数。再维护 $cn{t}_i$ 表示数字 $i$ 的个数。

对于 $3$ 操作，直接输出 $sum$ 即可。

对于 $1$ 操作，$cn{t}_x\leftarrow cn{t}_x+1$，如果加完后 $cn{t}_x$ 为 $1$，说明这是一个新的数字，$sum\leftarrow sum+1$。

对于 $2$ 操作，$cn{t}_x\leftarrow cn{t}_x-1$，如果减完后 $cn{t}_x$ 为 $0$，说明这个数字没有了，$sum\leftarrow sum-1$。

signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0); cout.tie(0);
	
	cin >> q;
	while (q--) {
		int opt, x; cin >> opt;
		if (opt == 3) cout << sum << '\n';
		else if (opt == 1) {
			cin >> x;
			cnt[x] ++;
			sum += (cnt[x] == 1);
		}
		else {
			cin >> x;
			cnt[x] --;
			sum -= (!cnt[x]);
		}
	}
	return 0;
}