专题四：并查集(dsu)

其他

1.HDU 1272 小希的迷宫

• 这题老坑了呀。亲测：
  （1）存在一组测试样例只输入两个0要额外判断。
  （2）编号不一定是1~n，存在编号跳跃的情况。
  （3）看清题啦，一定要保证图是连通的辣。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int maxn = 100005;
int par[maxn], ranking[maxn];
int N;

void init(int n);
int find(int x);
void unite(int x, int y);
bool same(int x, int y);

int main()
{
    int u, v;
    while(scanf("%d%d", &u, &v) && (u != -1) && (v != -1)){
        if(u == 0 && v == 0){
            printf("Yes\n");
            continue;
        }
        vector<int> vs;
        vs.push_back(u);
        vs.push_back(v);

        bool flag = 1;
        init(maxn);
        unite(u, v);

        while(scanf("%d%d", &u, &v) && u){
            if(same(u, v)){
                flag = 0;
            }
            unite(u, v);
            vs.push_back(u);
            vs.push_back(v);
        }
        int sz = vs.size();
        for(int i = 0; i < vs.size(); i++){
            if(!same(vs[0], vs[i])){
                flag = 0;
            }
        }
        if(flag){
            printf("Yes\n");
        }
        else{
            printf("No\n");
        }
    }
    return 0;
}
void init(int n)
{
    for(int i = 1; i <= n; i++){
        par[i] = i;
        ranking[i] = 0;
    }
}
int find(int x)
{
    if(par[x] == x){
        return x;
    }
    return par[x] = find(par[x]);
}
void unite(int x, int y)
{
    x = find(x);
    y = find(y);
    if(x == y){
        return;
    }
    if(ranking[x] < ranking[y]){
        par[x] = y;
    }
    else{
        par[y] = x;
        if(ranking[x] == ranking[y]){
            ranking[x]++;
        }
    }
}
bool same(int x, int y)
{
    return find(x) == find(y);
}

2.240. 食物链

这道题展示大雪菜的做法。反正有点不好理解。真不行再听一下。

#include<cstdio>
const int maxn = 50005;
int p[maxn], d[maxn];
int N, K;
void init(int N) {
	for (int i = 1; i <= N; i++) p[i] = i;
}
int find(int x) {
	if (p[x] == x) return x;
	else {
		int u = find(p[x]);
		d[x] += d[p[x]];
		return p[x] = u;
	}
}
int main() {
	scanf("%d%d", &N, &K);
	init(N);
	int res = 0;
	while (K--) {
		int t, a, b;
		scanf("%d%d%d", &t, &a, &b);
		int pa = find(a), pb = find(b);
		if (a > N || b > N) {
			res++;
			continue;
		}
		if (t == 1) {
			//pa == pb的意思是两者建立过联系。
			if (pa == pb && (d[a] - d[b]) % 3)  res++;
			//只有两者没建立过联系才可以到进到else if条件，否则不能改变之前的。
			else if(pa != pb){
				p[pa] = pb;
				d[pa] = d[b] - d[a];
				//这里是凑 d[a] + x - d[b] % 3 == 0, x就是d[pa], 解一下就出来了。
			}
		}
		else {
			if (pa == pb && (d[a] - d[b] - 1) % 3) res++;
			else if(pa != pb){
				p[pa] = pb;
				d[pa] = d[b] - d[a] + 1;
			}
		}
	}
	printf("%d\n", res);
}

3.排除危险

就是实验室老师让把化学物品装箱。如果箱子里任何n个化学物品，正好包含n种元素，那么将组成一个容易爆炸的混合物。

这个题可以看作是否形成一个环。形成环的话就是会爆炸的情况。两个物体之间每连一条边，就相当于多了一种物质。形成环的话恰好点的数目和边的数目相等，就爆炸了。

#include<cstdio>
#include<set>
using namespace std;
const int maxn = 100010;
int p[maxn];
void init(int N) {
	for (int i = 1; i <= N; i++) {
		p[i] = i;
	}
}
int find(int x) {
	if (p[x] == x) return x;
	return p[x] = find(p[x]);
}
int unite(int x, int y) {
	if (find(x) == find(y)) return 0;
	p[find(x)] = find(y);
	return 1;
}
int main() {
	init(100000);
	int a, b, cnt = 0, ans = 0;

	while (scanf("%d", &a) && a != -1) {
		scanf("%d", &b);
		if (!unite(a, b)) ans++;
	}
	printf("%d\n", ans);
	return 0;
}

4.1252. 搭配购买

• 并查集+01背包问题，不难。
• 有一个地方值得一提，就是判断是否为祖宗节点，若p[x] == x，那么x就是祖宗节点。反之则不是祖宗节点。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100010;
int p[maxn], v[maxn], w[maxn], N, M, W, f[maxn];
void init(int N) {
	for (int i = 1; i <= N; i++) p[i] = i;
}
int find(int x) {
	if (p[x] == x) return x;
	return p[x] = find(p[x]);
}
void unite(int a, int b) {
	if (find(a) == find(b)) return;
	v[find(b)] += v[find(a)];
	w[find(b)] += w[find(a)];
	p[find(a)] = find(b);
}
int main() {
	scanf("%d%d%d", &N, &M, &W);
	init(N);
	for (int i = 1; i <= N; i++) {
		scanf("%d%d", &w[i], &v[i]);
	}
	while (M--) {
		int a, b;
		scanf("%d%d", &a, &b);
		unite(a, b);
	}
	for (int i = 1; i <= N; i++) {
		if (p[i] != i) continue;
		for (int j = W; j >= w[i]; j--) {
			f[j] = max(f[j], f[j - w[i]] + v[i]);
		}
	}
	printf("%d\n", f[W]);
	return 0;
}

cf

1.B1. Books Exchange (easy version)

这个题讲的就是同学之间互相传递书籍。问多少天可以书可以回来。

在并查集里面，这个环的思想太重要了。这个就是，不过只是看看形成的环的大小，就是需要多少天书才会回来。

#include<cstdio>
int p[200005], sz[200005], a[200005];
void init(int N) {
	for (int i = 1; i <= N; i++) {
		p[i] = i;
		sz[i] = 1;
	}
}
int find(int x) {
	if (p[x] == x) return x;
	return p[x] = find(p[x]);
}
void unite(int x, int y) {
	if (find(x) == find(y)) return;
	sz[find(y)] += sz[find(x)];
	p[find(x)] = find(y);
}
int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		int N;
		scanf("%d", &N);
		init(N);
		for (int i = 1; i <= N; i++) {
			scanf("%d", &a[i]);
		}
		for (int i = 1; i <= N; i++) {
			unite(i, a[i]);
		}
		for (int i = 1; i <= N; i++) {
			printf("%d%c", sz[find(i)], i == N ? '\n' : ' ');
		}
	}
	return 0;
}

2.B. Drazil and His Happy Friends

这道题好容易错啊！这道题就是讲的是有N个男孩，M个女孩。然后第i天第i % N 个男孩可以和第 i % M 个女孩去约会，然后只要其中有一个人是happy的，他们就全变成happy的。

这道题可以这么抽象出来。给同一个集合的元素共同附加上一个属性。 他们中间只有有一个元素有这个属性，就把他加到这个集合上。这里展现出unite函数，一定要注意那个改变属性的位置。一定要先改变，在判断他们是不是同一集合。

void unite(int x, int y) {
	if (happy[x] || happy[y]) happy[find(y)] = 1;
	if (find(x) == find(y)) return;
	p[find(x)] = find(y);
}

但是，此题可以当作思维题去解。因为我们发现他们能否见面和N和M的最大公约数有关系。比如N为2，M为4时，有一些人就不能见面。因此只有 的时候（这里的g为最大公约数），所有满足的x，只要有一个人happy，那么所有的x都会happy。

因此，我们可以让x从0遍历到g - 1。然后看看N和M中有没有happy的。如果没有一个就输出No。具体来讲就是这样：

#include<cstdio>
#include<algorithm>
using namespace std;
int b[105], g[105];
int gcd(int a, int b) {
	if (b == 0) return a;
	return gcd(b, a % b);
}
int main() {
	int N, M, A, B, tmp;
	scanf("%d%d", &N, &M);
	scanf("%d", &A);
	for (int i = 0; i < A; i++) {
		scanf("%d", &tmp);
		b[tmp] = 1;
	}
	scanf("%d", &B);
	for (int i = 0; i < B; i++) {
		scanf("%d", &tmp);
		g[tmp] = 1;
	}
	int G = gcd(N, M);
	int flag = true;
	for (int x = 0; x < G; x++) {
		int ok = false;
		for (int i = x; i < N; i += G) {
			if (b[i]) {
				ok = true;
				break;
			}
		}
		for (int i = x; i < M; i += G) {
			if (g[i]) {
				ok = true;
				break;
			}
		}
		if (!ok) {
			flag = false;
			break;
		}
	}
	if (flag) printf("Yes\n");
	else printf("No\n");
	return 0;
}