并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合(换句换说每个节点都是一棵树),然后按照一定的顺序将属于同一组的元素合并,其间要反复查找一个元素在哪个元素集合中。
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入输出格式
输入格式:
第一行包含两个整数N、M,表示共有N个元素和M个操作。
接下来M行,每行包含三个整数Zi、Xi、Yi
当Zi=1时,将Xi与Yi所在的集合合并
当Zi=2时,输出Xi与Yi是否在同一集合内,是的话输出Y;否则话输出N
输出格式:
如上,对于每一个Zi=2的操作,都有一行输出,每行包含一个大写字母,为Y或者N
#include <iostream>
using namespace std;
int f[1000];
//用f数组存储当前k的祖先
int find(int k)
{
if (f[k] == k)
return k;
return f[k] = find(f[k]);
}
int main()
{
int n, m;
cin >> n >> m;
for (int i = 1; i <= 1000; ++i)
{
//初始化他们的祖先都为本身
f[i] = i;
}
int x, y, z;
for (int i = 0; i < m; ++i)
{
cin >> z >> x >> y;
if (z == 1)
{
f[x] = find(y);
}
if (z == 2)
{
if (find(x) == find(y))
{
cout << "Y" << endl;
}
else
cout << "N" << endl;
}
}
}
修路问题
题目背景
AA地区在地震过后,连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。
题目描述
给出A地区的村庄数NN,和公路数MM,公路是双向的。并告诉你每条公路的连着哪两个村庄,并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车,即最早什么时候任意两条村庄都存在至少一条修复完成的道路(可以由多条公路连成一条道路)
输入输出格式
输入格式:
第11行两个正整数N,MN,M
下面MM行,每行33个正整数x, y, tx,y,t,告诉你这条公路连着x,yx,y两个村庄,在时间t时能修复完成这条公路。
输出格式:
如果全部公路修复完毕仍然存在两个村庄无法通车,则输出-1−1,否则输出最早什么时候任意两个村庄能够通车。
输入输出样例
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int f[1000];
struct Node
{
int x, y, z;
};
bool comp(Node a, Node b)
{
return a.z < b.z;
}
int find(int k)
{
if (k == f[k])
return k;
return f[k] = find(k);
}
int main()
{
int N, M;
cin >> N >> M;
for (int i = 0; i <= N; ++i)
f[i] = i;
vector<Node> nvec(M + 1);
for (int i = 1; i <= M; ++i)
{
cin >> nvec[i].x >> nvec[i].y >> nvec[i].z;
}
sort(nvec.begin() + 1, nvec.end(), comp);
int val = 2;
f[nvec[1].y] = nvec[1].x;
int max = nvec[1].z;
for (int i = 2; i < nvec.size(); ++i)
{
if (val == N)
break;
if (find(nvec[i].x) == find(nvec[i].y))
continue;
else
{
f[nvec[i].y] = f[nvec[i].x];
val++;
if (max < nvec[i].z)
max = nvec[i].z;
}
}
if (val < N)
cout << -1;
else
cout << max;
}