CF1559D1. Mocha and Diana (Easy Version)

原题链接:1559D1. Mocha and Diana (Easy Version)

题意:

小明和小红各有一个具有\(n\)个结点的森林,现执行操作:

  • 加一条边,使得两人的森林还是森林
  • 小明加一条\((u, v)\)的边,那么小红也必须加一条\((u, v)\)的边。
    问我们最多能加多少边?

思路:

很明显,第一个条件没啥用,关键是第二个条件,我们知道如果一个人不能加\((u, v)\)一条边的前提条件是\(u\)与\(v\)已经联通,那么根据这个条件,我们可以使用并查集,而边的数据范围是\([1-1000]\),所以可以直接暴力枚举点,然后使用并查集来判断两个点是否已经在一个集合内,如果在一个集合内那么就不能加边了,否则加上即可。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;
int fa1[N], fa2[N];

int find1(int x) {
	if (x != fa1[x]) fa1[x] = find1(fa1[x]);
	return fa1[x];
}

int find2(int x) {
	if (x != fa2[x]) fa2[x] = find2(fa2[x]);
	return fa2[x];
}

int main() {
	ios::sync_with_stdio(false), cin.tie(0);
	
	int n, m1, m2;
	cin >> n >> m1 >> m2;
	for (int i = 1; i <= n; i++) fa1[i] = i, fa2[i] = i;
	while (m1--) {
		int a, b;
		cin >> a >> b;
		fa1[find1(a)] = find1(b);
	}
	
	while (m2--) {
		int a, b;
		cin >> a >> b;
		fa2[find2(a)] = find2(b);
	}
	
	vector<pair<int, int>> add;
	for (int i = 1; i <= n; i++) {
		for (int j = i + 1; j <= n; j++) {
			int u1 = find1(i), v1 = find1(j);
			int u2 = find2(i), v2 = find2(j);
			if (u1 != v1 && u2 != v2) {
				add.push_back({i, j});
				fa1[u1] = v1;
				fa2[u2] = v2;	
			}
		}
	}
	
	cout << add.size() << endl;
	for (int i = 0; i < add.size(); i++) 
		cout << add[i].first << " " << add[i].second << endl;
	
    return 0;
}

注意:

题目思路并不难,但是我耗了一个多小时,原因:
我一开始\(find\)函数写成了这样。

int find1(int x) {
	if (x == fa1[x]) return x;
	return find1(fa1[x]);
}

乍一看没错,确实没错,就是\(T\)到飞起,为啥,因为没有加入路径压缩的优化,所以\(T\)到飞起,我服了,这次长记性了。
另外,以后并查集用一个类吧,贴个板子:

struct DSU{
	int fa[N], Size[N];
	void init() {
		for (int i = 1; i <= n; i++) fa[i] = i;
	}
	int find(int x) {
		if (x != fa[x]) fa[x] = find(fa[x]);
		return fa[x];
	}
	void merge(int x, int y) {
		Size[find(x)] += Size[find(y)];
		fa[find(x)] = find(y);
	}
	bool check(int x, int y) {
		return find(x) == find(y);
	}
}dsu1, dsu2;
上一篇:调和


下一篇:BZOJ 3450 - Tyvj1952 Easy (期望DP)