营救(并查集 + 图论 + prim/Kruskal + 最小生成树)

原题链接

题目背景

“咚咚咚……”“查水表！”原来是查水表来了，现在哪里找这么热心上门的查表员啊！小明感动得热泪盈眶，开起了门……

题目描述

妈妈下班回家，街坊邻居说小明被一群陌生人强行押上了警车！妈妈丰富的经验告诉她小明被带到了t区，而自己在 s区。该市有 m条大道连接 n个区，一条大道将两个区相连接，每个大道有一个拥挤度。小明的妈妈虽然很着急，但是不愿意拥挤的人潮冲乱了她优雅的步伐。所以请你帮她规划一条从 s至 t的路线，使得经过道路的拥挤度最大值最小。

输入格式

第一行有四个用空格隔开的 n，m，s，t，其含义见【题目描述】。接下来 m行，每行三个整数 u, v, w，表示有一条大道连接区 uu 和区 v，且拥挤度为 w。两个区之间可能存在多条大道。

输出格式

输出一行一个整数，代表最大的拥挤度。

输入样例

3 3 1 3
1 2 2
2 3 1
1 3 3

输出样例

2

分析

• 题意就是选择一条路线可以将s和t联通，这条路线中最大的权值要尽可能地小。
• 分析易得，本题应使用kruskal算法构造最小生成树：因为最小生成树可以保证一条路径的最大值最小 —— 每次选择一条权值最小的边加入集合，当集合连通时加入的边的权值即为答案。
• Kruskal算法每次选取权值最小的边加入集合,因此要先将边按照权值 / 拥挤度排序。
• 在本题中，并查集的作用主要是维护连通性， 防止Kruskal算法中产生了回路。

AC代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10010;
const int M = 20010;
int parent[N];//disjoint set

struct edge
{
    int u;//block u
    int v;//block v
    int w;//weight 
}nums[M];

bool cmp(edge u,edge v)
{
    return u.w < v.w;
}

int find(int x)//disjoint set's find + compression
{
    if (x != parent[x]) parent[x] = find(parent[x]);
    return parent[x];
}

int Union(int x,int y)//disjoint set's union
{
    int a = find(x);
    int b = find(y);
    if (a != b) 
    {
        parent[a] = b;
        return true;//union successfully
    }
    return false;//union false
}

int main()
{
    int n,m,s,t;
    while (cin >> n >> m >> s >> t)
    {
        for (int i = 1;i <= n;i++) parent[i] = i;//init disjoint set
        
        for (int i = 1;i <= m;i++) cin >> nums[i].u >> nums[i].v >> nums[i].w;//read data
        
        sort(nums + 1,nums + m + 1,cmp);//edge : from small to big
        
        //Kruskal algorithm
        for (int i = 1;i <= m;i++)
        {
            if (find(nums[i].u) != find(nums[i].v))//u、v两个街区不在同一集合时 
            {
                Union(nums[i].u,nums[i].v);//合并这两个集合(即选择权值最小边加入集合)
                
                if (find(s) == find(t))//若s和t在某次合并集合之后连通
                {
                    cout << nums[i].w << endl;//输出此时的权值
                    break;
                }
            }
        }
    }
    
    return 0;
}