原题链接: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;