C语言-算法-并查集

SpongeG

已于 2024-02-27 16:47:52 修改

阅读量632

点赞数 9

文章标签：算法 c语言开发语言

于 2024-01-23 16:31:11 首次发布

本文链接：https://blog.csdn.net/2301_79837864/article/details/135774846

版权

【模板】并查集

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

第一行包含两个整数 $N, M$ ,表示共有 $N$ 个元素和 $M$ 个操作。

接下来 $M$ 行，每行包含三个整数 $Z_i,X_i,Y_i$ 。

当 $Z_i=1$ 时，将 $X_i$ 与 $Y_i$ 所在的集合合并。

当 $Z_i=2$ 时，输出 $X_i$ 与 $Y_i$ 是否在同一集合内，是的输出
Y ；否则输出 N 。

输出格式

对于每一个 $Z_i=2$ 的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N 。

样例 #1

样例输入 #1

样例输出 #1

N
Y
N
Y

提示

对于 $30\%$ 的数据， $\le 10$ ， $\le 20$ 。

对于 $70\%$ 的数据， $\le 100$ ， $\le 10^3$ 。

对于 $100\%$ 的数据， $1\le N \le 10^4$ ， $1\le M \le 2\times 10^5$ ， $\le X_i, Y_i \le N$ ， $Z_i \in \{ 1, 2 \}$ 。

代码实现

#include <stdio.h>
#include <stdlib.h>
void init(int n); // 初始化并查集
int find(int x); // 查找元素x所在的集合，路径压缩
void un(int x, int y); // 合并元素x和元素y所在的集合
#define MAX 300000
int pa[MAX]; // 表示元素i的父节点

int main(int argc, char *argv[])
{
	int N, M, Z, X, Y, i;
	scanf("%d %d", &N, &M);
	init(N); // 初始化并查集
	
	for (i = 0; i < M; i++)
	{
		scanf("%d %d %d", &Z, &X, &Y);
		if (Z == 1)
		{
			un(X, Y); // 合并
		}
		else
		{
			if (find(X) == find (Y))
			{
				printf("Y\n");
			}
			else
			{
				printf("N\n");
			}
		}
	}
	return 0;
}

void init(int n) // 初始化并查集
{
	int i;
	for (i = 1; i <= n; i++)
	{
		pa[i] = i;
	}
}

int find(int x) // 查找元素x所在的集合，路径压缩
{
	if (pa[x] != x)
	{
		pa[x] = find(pa[x]);
	}
	return pa[x];
}

void un(int x, int y) // 合并元素x和元素y所在的集合
{
	int rootX = find(x);
	int rootY = find(y);
	if (rootX != rootY)
	{
		pa[rootX] = rootY;
	}
}

修复公路

题目背景

A 地区在地震过后，连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。

题目描述

给出 A 地区的村庄数 $N$ ，和公路数 $M$ ，公路是双向的。并告诉你每条公路的连着哪两个村庄，并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车，即最早什么时候任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路）。

输入格式

第 $1$ 行两个正整数 $N, M$ 。

下面 $M$ 行，每行 $3$ 个正整数 $x, y, t$ ，告诉你这条公路连着 $x, y$ 两个村庄，在时间t时能修复完成这条公路。

输出格式

如果全部公路修复完毕仍然存在两个村庄无法通车，则输出 $- 1$ ，否则输出最早什么时候任意两个村庄能够通车。

样例 #1

样例输入 #1

样例输出 #1

提示

$1\leq x, y\leq N \le 10 ^ 3$ ， $1\leq M, t \le 10 ^ 5$ 。

代码实现

#include <stdio.h>
#include <stdlib.h>
int find(int x); // 并查集的查找操作
void un(int x, int y); // 合并元素x和元素y所在的集合
int cmp(const void *a, const void *b); // 边的比较函数，用于排序
#define MAX 100000
int pa[MAX]; // 并查集的数据结构，表示元素i的父节点
typedef struct // 边的数据结构
{
    int u, v, w;
} Edge;
Edge edge[MAX];

int main()
{
    int N, M, i, maxW;
    scanf("%d %d", &N, &M);
    maxW = -1;
    
    for (i = 1; i <= N; i++) // 初始化并查集
    {
        pa[i] = i;
    }

    for (i = 0; i < M; i++)
    {
        scanf("%d %d %d", &edge[i].u, &edge[i].v, &edge[i].w);
    }
    qsort(edge, M, sizeof(Edge), cmp); // 按照边的权重排序
    for (i = 0; i < M; i++)
    {
        if (find(edge[i].u) != find(edge[i].v))
        {
            un(edge[i].u, edge[i].v);
            maxW = edge[i].w;
        }
    }
    for (i = 2; i <= N; i++)
    {
        if (find(i) != find(1))
        {
            printf("-1\n");
            return 0;
        }
    }
    printf("%d\n", maxW);

    return 0;
}

int find(int x) // 并查集的查找操作
{
    if (pa[x] != x)
    {
        pa[x] = find(pa[x]);
    }
    return pa[x];
}

void un(int x, int y) // 合并元素x和元素y所在的集合
{
    int rootX = find(x);
    int rootY = find(y);
    if(rootX != rootY)
    {
        pa[rootX] = rootY;
    }
}

int cmp(const void *a, const void *b) // 边的比较函数，用于排序
{
    return ((Edge *)a) -> w - ((Edge *)b) -> w;
}

亲戚

题目背景

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

题目描述

规定： $x$ 和 $y$ 是亲戚， $y$ 和 $z$ 是亲戚，那么 $x$ 和 $z$ 也是亲戚。如果 $x$ ， $y$ 是亲戚，那么 $x$ 的亲戚都是 $y$ 的亲戚， $y$ 的亲戚也都是 $x$ 的亲戚。

输入格式

第一行：三个整数 $n, m, p$ ，（ $\le 5000$ ），分别表示有 $n$ 个人， $m$ 个亲戚关系，询问 $p$ 对亲戚关系。

以下 $m$ 行：每行两个数 $M_i$ ， $M_j$ ， $\le M_i,~M_j\le n$ ，表示 $M_i$ 和 $M_j$ 具有亲戚关系。

接下来 $p$ 行：每行两个数 $P_i,P_j$ ，询问 $P_i$ 和 $P_j$ 是否具有亲戚关系。

输出格式

$p$ 行，每行一个 Yes 或 No。表示第 $i$ 个询问的答案为“具有”或“不具有”亲戚关系。

样例 #1

样例输入 #1

样例输出 #1

Yes
Yes
No

代码

#include <stdio.h>
int find(int x); // 查询
void un(int x, int y); // 合并
#define MAX 10000
int pa[MAX]; // i元素的父节点

int main()
{
    int n, m, p, i, Mi, Mj, Pi, Pj;
    scanf("%d %d %d", &n, &m, &p); // n个人，m个亲戚关系，p对亲戚关系

    for (i = 1; i <= n; i++) // 初始化并查集
    {
        pa[i] = i;
    }
    for (i = 1; i <= m; i++) // 具有亲戚关系
    {
        scanf("%d %d", &Mi, &Mj);
        if (find(Mi) != find(Mj)) // 两个节点的父节点不同
        {
            un(Mi, Mj);
        }
    }
    for (i = 1; i <= p; i++) // 询问是否具有亲戚关系
    {
        scanf("%d %d", &Pi, &Pj);
        if (find(Pi) == find(Pj)) // 父节点相同，则具有亲戚关系
        {
            printf("Yes\n");
        }
        else
        {
            printf("No\n");
        }
    }
    return 0;
}

int find(int x) // 找到元素x的根节点
{
    if (pa[x] != x)
    {
        pa[x] = find(pa[x]);    
    }
    return pa[x];
}

void un(int x, int y) // 合并元素x和元素y所在的集合
{
    int rootX = find(x);
    int rootY = find(y);
    if (rootX != rootY)
    {
        pa[rootX] = rootY;
    }
}

一中校运会之百米跑

题目背景

在一大堆秀恩爱的 ** 之中，来不及秀恩爱的苏大学神踏着坚定（？）的步伐走向了 $100$ 米跑的起点。这时苏大学神发现，百米赛跑的参赛同学实在是太多了，连体育老师也忙不过来。这时体育老师发现了身为体育委员的苏大学神，便来找他帮忙。

可是苏大学神需要热身，不然跑到一半就会抽（筋）、于是他就找到了你。。。如果你帮助体育老师解决了问题，老师就会给你 $5$ 个积分。

题目描述

假设一共有 $N$ （ $2\leq N\leq 2\times 10^4$ ）个参赛选手。（尼玛全校学生都没这么多吧）

老师会告诉你这 $N$ 个选手的名字。

接着会告诉你 $M$ （ $1\leq M\leq 10^6$ ）句话，即告诉你学生 A 与学生 B 在同一个组里。

如果学生 A 与学生 B 在同一组里，学生 B 与学生 C 也在同一组里，就说明学生 A 与学生 C 在同一组。

然后老师会问你 $K$ （ $1\leq K\leq 10^6$ ）句话，即学生 X 和学生 Y 是否在同一组里。

若是则输出 Yes.，否则输出 No.。

输入格式

第一行输入 $N$ 和 $M$ 。

接下来 $N$ 行输入每一个同学的名字。

再往下 $M$ 行每行输入两个名字，且保证这两个名字都在上面的 $N$ 行中出现过，表示这两个参赛选手在同一个组里。

再来输入 $K$ 。

接下来输入 $K$ 个体育老师的询问。

输出格式

对于每一个体育老师的询问，输出 Yes. 或 No.。

样例 #1

样例输入 #1

10 6
Jack
Mike
ASDA
Michel
brabrabra
HeHe
HeHE
papapa
HeY
Obama
Jack Obama
HeHe HeHE
brabrabra HeHe
Obama ASDA
papapa Obama
Obama HeHE
3
Mike Obama
HeHE Jack
papapa brabrabra

样例输出 #1

No.
Yes.
Yes.

代码实现

#include <stdio.h>
#include <string.h>
#define MAX 3000
int pa[MAX];
int find(int x);
void un(int x, int y);

int main()
{
    int N, M, K, i, j;
    char name[MAX][20], Mi[MAX], Mj[MAX], Ki[MAX], Kj[MAX];
    scanf("%d %d", &N, &M);

    for (i = 1; i <= N; i++)
    {
        scanf("%s", name[i]);
    }
    for (i = 1; i <= N; i++)
    {
        pa[i] = i;
    }
    for (i = 1; i <= M; i++)
    {
        scanf("%s %s", Mi, Mj);

        int x = -1, y = -1;
        for (j = 1; j <= N; j++)
        {
            if (strcmp(name[j], Mi) == 0)
            {
                x = j;
                break;
            }
        }
        for (j = 1; j <= N; j++)
        {
            if (strcmp(name[j], Mj) == 0)
            {
                y = j;
                break;
            }
        }
        int rootX = find(x); // 获取根节点
        int rootY = find(y);
        if (find(x) != find(y))
        {
            un(x, y);
        }
    }
    scanf("%d", &K);
    for (i = 1; i <= K; i++)
    {
        scanf("%s %s", Ki, Kj);

        int x = -1, y = -1;
        for (j = 1; j <= N; j++)
        {
            if (strcmp(name[j], Ki) == 0)
            {
                x = j;
                break;
            }
        }
        for (j = 1; j <= N; j++)
        {
            if (strcmp(name[j], Kj) == 0)
            {
                y = j;
                break;
            }
        }
        int rootX = find(x); // 获取根节点
        int rootY = find(y);
        if (rootX != rootY)
        {
            printf("No.\n");
        }
        else
        {
            printf("Yes.\n");
        }
    }

    return 0;
}

int find(int x)
{
    if (pa[x] != x)
    {
        pa[x] = find(pa[x]);
    }
    return pa[x];
}

void un(int x, int y)
{
    int rootX = find(x);
    int rootY = find(y);
    if (rootX != rootY)
    {
        pa[rootX] = rootY;
    }
}

SpongeG

关注

9
点赞
踩
8

收藏

觉得还不错? 一键收藏
0
评论
C语言-算法-并查集

对于100%的数据，1≤N≤10^4， 1≤M≤2×10^51≤Xi ,Yi≤N，Zi∈{1,2}。对于每一个的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N。如题，现在有一个并查集，你需要完成合并和查询操作。第一行包含两个整数 ,表示共有个元素和个操作。对于70%的数据，N≤100，M≤10^3。当时，输出与是否在同一集合内，是的输出。对于30%的数据，N≤10，M≤20。接下来行，每行包含三个整数。当时，将与所在的集合合并。
复制链接

扫一扫