【并查集】并查集例题讲解 洛谷 P2914 家谱 含代码解析

并查集例题讲解 洛谷P2814 含代码讲解

并查集介绍

• 并查集是一种数据结构和算法，用于处理“查找”和“合并”两个基本操作的问题。它通常应用于图论、集合论等领域，特别是在解决“连通性”问题上，比如找出两个元素是否在同一集合或分组内。在并查集中，每个元素都属于一个集合，通过根节点来进行标识。
• 其工作原理是维护一个数组或哈希表，其中索引表示元素，值指向元素的根节点。当需要查找两个元素是否属于同一集合时（查找），首先分别找到它们的根节点，如果它们指向相同的节点，则说明它们在同一集合；若不同，则将其中一个的根节点更新为其另一个的根节点（合并）。这样可以保持树状结构，查找和合并的时间复杂度一般为O(logn)。

并查集例题

• 洛谷 P2814 家谱 👇

家谱

题目背景

现代的人对于本家族血统越来越感兴趣。

题目描述

给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入格式

输入由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系中父亲只有一行，儿子可能有若干行，用 #name 的形式描写一组父子关系中的父亲的名字，用 +name 的形式描写一组父子关系中的儿子的名字；接下来用 ?name 的形式表示要求该人的最早的祖先；最后用单独的一个 $ 表示文件结束。

输出格式

按照输入文件的要求顺序，求出每一个要找祖先的人的祖先，格式为：本人的名字 $+$ 一个空格 $+$ 祖先的名字 $+$ 回车。

样例 #1

样例输入 #1

#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$

样例输出 #1

Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

提示

规定每个人的名字都有且只有 $6$ 个字符，而且首字母大写，且没有任意两个人的名字相同。最多可能有 $10^3$ 组父子关系，总人数最多可能达到 $5\times 10^4$ 人，家谱中的记载不超过 $30$ 代。

代码加讲解

根据题目看出，第一步读取每一行的第一个字符来进行相应的操作，第二步将父亲与儿子连接起来，第三步就是向上找到祖先了。而这道题最大的不同点在于它的集合元素都是字符串，所以我们一般用map加上string的返回值来进行处理。但如果元素是整型的话还是应开数组进行连接，否则容易超时。具体步骤：

1. 取出第一个字符和后面的字符串
2. 判断字符进行连接
3. 根据输入内容来输出

代码演示

#include<bits/stdc++.h>
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
const int N = 1e3+1;
string fa;//定义全局变量来存储父结点的字符串
map <string,string> mp;//用map来进行字符串的“连接”

string find(string x)//一个string返回值的递归找祖函数
{
	if(x != mp[x])		mp[x] = find(mp[x]);//自己不对应自己时，表明x不是祖先，就继续递归
	return mp[x];//返回祖先的值
}


int main()
{
	IOS;//加快输入输出，同时将输入和输出分开，各为一个整体
	while(1)//先写个死循环
	{
		char a;
		string x;
		cin>>a;//读入第一个字符
		if(a  == '$')	return 0;//判断是否要终止程序
		cin>>x;//继续读入另一个字符
		if(mp.count(x) == 0)	mp[x] = x;//当这个人名还没被连接的时候，进行初始化，即map对应自己
		if(a == '#')//若是父结点，则存入全局变量，以供下次使用
		{
			fa = x;
		}
		if(a == '+')//若是子结点
		{
			if(find(x) != find(fa))		mp[find(x)] = find(fa);//向上递归到根，若不在同一个集合内，就将父子连接起来
		}
		if(a == '?')//判断是否要输出
		{
			cout<<x<<" "<<find(x)<<"\n";//输出名字和他的祖宗
		}
	}
}

总结

此题是一道非常经典的并查集题，利用map代替数组进行集合的连接，顺利解决集合对应问题，利用组成集合并递归map来查找祖先。