洛谷

洛谷题单指南-最小生成树-P1396 营救

原题链接:https://www.luogu.com.cn/problem/P1396

题意解读:找到一条从s到t的路径,使得最大边权值最小,求这个最大的边权值。

解题思路:

根据kruskal的算法模型,对边权值小到大处理的过程,如果加入某条边权后,出现s、t连通,那么最后一次的边权即s到t路径上的最大边权最小值。

证明:如果存在s到t路径上更小的最大边权,那么这样的生成树的权值比kruskal算法生成树的权值更小,而kruskal算法生成的树就是最小生成树,产生矛盾。

100分代码:

#include <bits/stdc++.h>
using namespace std;

const int N = 10004, M = 20005;
struct Edge
{
    int u, v, w;
} edges[M];
int p[N];
int n, m, s, t;

bool cmp(Edge a, Edge b)
{
    return a.w < b.w;
}

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

int kruskal()
{
    sort(edges + 1, edges + m + 1, cmp);
    for(int i = 1; i <= n; i++) p[i] = i;
    for(int i = 1; i <= m; i++)
    {
        int a = find(edges[i].u);
        int b = find(edges[i].v);
        if(a != b)
        {
            p[a] = b;
            if(find(s) == find(t)) return edges[i].w;
        }
    }
}

int main()
{
    cin >> n >> m >> s >> t;
    for(int i = 1; i <= m; i++) 
    {
        cin >> edges[i].u >> edges[i].v >> edges[i].w;
    }
    cout << kruskal();
    return 0;
}