一、前言
接(1)完成剩余题目和了解并查集运用求解最小生成树的Kruskal算法
二、专题训练
2.1 题目总览
前四题见(1)
2.2 1568: 并查集-家谱
思路
首先这个题目的描述就有问题,它说每一组的父子关系由两行组成,那样例的第3到第5行你要怎么解释,麻了。这里应该是#开头的是父亲,下面紧跟着的若干行+开头的都是他的儿子。而且这个题也是不适合用我们的模板去解的,因为它需要从字符串映射到字符串,当然你也可以给字符串编个序号这样也可以实现,我这里给出另一种解法,我们用哈希表来实现字符串到字符串的映射。其余的就是并查集的基本操作了
参考代码
#include <bits/stdc++.h>
#include <functional>
using i64 = long long;
using pss = std::pair<std::string,std::string>;
int main() {
std::cin.tie(nullptr)->sync_with_stdio(false);
std::vector<pss> v;
pss pair = {"",""};
std::vector<std::string> query;
std::string line;
std::unordered_map<std::string,std::string> fa;
std::string father,son;
using Find = std::function<std::string(std::string)>;
Find find = [&](std::string x) {
if(x!=fa[x]) fa[x] = find(fa[x]);
return fa[x];
};
while(std::getline(std::cin,line)) {
if(line[0]=='#') {
father = line.substr(1);
if(fa[father]=="") fa[father] = father;
}else if(line[0]=='+') {
son = line.substr(1);
if(fa[son]=="") {
fa[son] = find(father);
}
}else if(line[0]=='?') {
query.emplace_back(line.substr(1));
}else {//line[0]=='$'
break;
}
}
for(auto &str:query) {
std::cout << str << ' ' << find(str) << '\n';
}
return 0;
}
2.3 1836: 并查集-格子游戏
思路
对于并查集来说,一个节点的值最常见的是一个int,上一题节点的值是一个字符串,这道题节点的值是一个坐标点,当然你可以像上一题一样,写一个坐标的结构体,然后用哈希表将坐标映射到坐标,我这里采用将坐标变换成一个int,然后每次操作时判断,要连接的两个点是否已经在同一个并查集中了,如果是则输出当前的步数,如果不是则将两个节点进行合并,执行到结束,如果游戏仍然没有结束则输出"draw"
参考代码
注意并查集初始化的时候,空间要开大一些
#include <bits/stdc++.h>
using i64 = long long;
struct DisjointSet {
int _n;
std::vector<int> _fa,_size;
DisjointSet(){}
DisjointSet(int n){
init(n);
}
void init(int n) {
_fa.resize(n);
std::iota(_fa.begin(),_fa.end(),0);
_size.assign(n,1);
}
int find(int x) {
if(x!=_fa[x]) {
_fa[x] = find(_fa[x]);
}
return _fa[x];
}
bool same(int x,int y) {
return find(x)==find(y);
}
bool merge(int x,int y) {
int fx = find(x);
int fy = find(y);
if(fx!=fy) {
_size[fx]+=_size[fy];
_fa[fy] = fx;
return true;
}
return false;
}
};
int main() {
std::cin.tie(nullptr)->sync_with_stdio(false);
int n,m;
std::cin >> n >> m;
DisjointSet disjointSet = DisjointSet((n+5)*(n+2));
for(int i = 0;i<m;i++) {
int x1,y1,x2,y2;
std::cin >> x1 >> y1;
std::string op;
std::cin >> op;
if(op=="D") {
x2 = x1+1,y2 = y1;
}else {
x2 = x1,y2 = y1+1;
}
int t1 = n*(x1-1)+y1-1,t2 = n*(x2-1)+y2-1;
if(!disjointSet.merge(t1,t2)) {
std::cout << i+1 << '\n';
return 0;
}
}
std::cout << "draw" << '\n';
return 0;
}
2.4 1837: 并查集-亲戚
思路
此题思路很一般,正常做就行,但是因为数据量的限制,我们写并查集的时候一定要做路径压缩,不然会TLE,直接用我们的模板就行
参考代码
#include <bits/stdc++.h>
using i64 = long long;
struct DisjointSet {
int _n;
std::vector<int> _fa,_size;
DisjointSet(){}
DisjointSet(int n){
init(n);
}
void init(int n) {
_fa.resize(n);
std::iota(_fa.begin(),_fa.end(),0);
_size.assign(n,1);
}
int find(int x) {
if(x!=_fa[x]) {
_fa[x] = find(_fa[x]);
}
return _fa[x];
}
bool same(int x,int y) {
return find(x)==find(y);
}
bool merge(int x,int y) {
int fx = find(x);
int fy = find(y);
if(fx!=fy) {
_size[fx]+=_size[fy];
_fa[fy] = fx;
return true;
}
return false;
}
};
int main() {
std::cin.tie(nullptr)->sync_with_stdio(false);
int n,m;
std::cin >> n >> m;
DisjointSet disjointSet = DisjointSet(n+1);
for(int i = 0;i<m;i++) {
int x,y;
std::cin >> x >> y;
disjointSet.merge(x,y);
}
int q;
std::cin >> q;
while(q--) {
int x,y;
std::cin >> x >> y;
std::cout << (disjointSet.same(x,y)?"Yes\n":"No\n");
}
return 0;
}
三、并查集的运用
3.1 最小生成树
最小生成树是一个现实中经常遇到的问题,各个城镇直接修路、修桥,我们力争使用的材料或者经费最少,就需要求出最小生成树
3.2 Prim算法(基于贪心算法,与求最短路的Dijkstra算法类似)
3.3 Kruskal算法(也是基于贪心算法,需要用到并查集)
把带权边按照权值升序排序,开始循环找当前权值最小的边,如果两个节点不在同一个并查集中,则将它们合并,如果在同一个并查集中,则继续找下一条边。循环结束,如果有n-1条边,则成功找到最小生成树(最小生成树不唯一,但最后的权值和最小是唯一的),如果循环结束,找不到n-1条边,则不存在最小生成树
3.3.1 例题1 AcWing 859. Kruskal算法求最小生成树
思路
没啥好说的,是一个模板题
参考代码
#include <bits/stdc++.h>
using i64 = long long;
struct Edge {
int _x, _y, _w;
Edge(int x, int y, int w) : _x(x), _y(y), _w(w) {}
bool operator<(const Edge& edge2) const {
return _w < edge2._w;
}
};
struct DisjointSet {
int _n;
std::vector<int> _fa,_size;
DisjointSet(){}
DisjointSet(int n){
init(n);
}
void init(int n) {
_fa.resize(n);
std::iota(_fa.begin(),_fa.end(),0);
_size.assign(n,1);
}
int find(int x) {
if(x!=_fa[x]) {
_fa[x] = find(_fa[x]);
}
return _fa[x];
}
bool same(int x,int y) {
return find(x)==find(y);
}
bool merge(int x,int y) {
int fx = find(x);
int fy = find(y);
if(fx!=fy) {
_size[fx]+=_size[fy];
_fa[fy] = fx;
return true;
}
return false;
}
};
int main() {
std::cin.tie(nullptr)->sync_with_stdio(false);
int n, m;
std::cin >> n >> m;
std::vector<Edge> edges;
DisjointSet disjointSet(n);
for (int i = 0; i < m; ++i) {
int x, y, w;
std::cin >> x >> y >> w;
edges.emplace_back(x, y, w);
}
std::sort(edges.begin(), edges.end());
int ans = 0, cnt = 0;
for (const auto& edge : edges) {
if (disjointSet.merge(edge._x, edge._y)) {
ans += edge._w;
++cnt;
if (cnt == n - 1) break;
}
}
std::cout << (cnt < n - 1 ? "impossible" : std::to_string(ans)) << '\n';
return 0;
}
3.3.2 例题2 最小生成树-最优布线问题
思路
题目的输入是邻接矩阵的形式,我们把它转换成边存储,然后再用Kruskal算法即可,当然这个题用Prim算法更合适
参考代码1(Kruskal算法)
#include <bits/stdc++.h>
using i64 = long long;
struct Edge {
int _x,_y,_w;
Edge(int x,int y,int w):_x(x),_y(y),_w(w){}
bool operator < (const Edge& edge2) const {
return _w<edge2._w;
}
};
struct DisjointSet {
int _n;
std::vector<int> _fa,_size;
DisjointSet(){}
DisjointSet(int n){
init(n);
}
void init(int n) {
_fa.resize(n);
std::iota(_fa.begin(),_fa.end(),0);
_size.assign(n,1);
}
int find(int x) {
if(x!=_fa[x]) {
_fa[x] = find(_fa[x]);
}
return _fa[x];
}
bool same(int x,int y) {
return find(x)==find(y);
}
bool merge(int x,int y) {
int fx = find(x);
int fy = find(y);
if(fx!=fy) {
_size[fx]+=_size[fy];
_fa[fy] = fx;
return true;
}
return false;
}
};
int main() {
std::cin.tie(nullptr)->sync_with_stdio(false);
int n;
std::cin >> n;
std::vector<Edge> edges;
DisjointSet disjointSet = DisjointSet(n+1);
for(int i = 1;i<=n;i++) {
for(int j = 1;j<=n;j++) {
int w;
std::cin >> w;
if(j>=i) {
edges.emplace_back(i,j,w);
}
}
}
std::sort(edges.begin(),edges.end());
int ans = 0;
for(int i = 0;i<edges.size();i++) {
int x = edges[i]._x,y = edges[i]._y,w = edges[i]._w;
int fx = disjointSet.find(x),fy = disjointSet.find(y);
if(disjointSet.merge(fx,fy)) {
ans+=w;
}
}
std::cout << ans << '\n';
return 0;
}
参考代码2(Prim算法)
#include <iostream>
#include <cstring>
#include <type_traits>
template<typename T1,typename T2>
typename std::common_type<T1,T2>::type min(T1 num1,T2 num2){
return num1>num2?num2:num1;
}
int main(){
std::cin.tie(nullptr)->sync_with_stdio(false);
constexpr int MAX_N = 1e2+5,INF = 0x3f3f3f3f;
int g[MAX_N][MAX_N];
int dist[MAX_N];
bool vis[MAX_N] {};
int n;std::cin >> n;
for(int i = 1;i<=n;++i){
for(int j = 1;j<=n;++j){
std::cin >> g[i][j];
}
}
auto prim = [&]()->int{
memset(dist,0x3f,sizeof dist);
int res = 0;
dist[1] = 0;
for(int i = 0;i<n;++i){
int t = -1;
for(int j = 1;j<=n;++j){
if(!vis[j]&&(t==-1||dist[j]<dist[t])) t = j;
}
if(dist[t]==INF) return INF;
vis[t] = true;
res+=dist[t];
for(int j = 1;j<=n;++j) dist[j]=min(dist[j],g[t][j]);
}
return res;
};
std::cout << prim() << '\n';
return 0;
}