poj1797 Heavy Transportation kruskal+并查集

题目大意是找到1到n的所有路径中权值最小边的最大值。

最小生成树kruskal并对并查集用启发式优化(按秩合并)。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cctype>
using namespace std;
struct node{
    int u;
    int v;
    int w;
};
node e[1000005];
int p[1005];
int r[1005];
int n,m;
bool cmp(node x,node y)
{
    return x.w>y.w;
}
int Find(int u)
{
    if(u!=p[u])
    p[u]=Find(p[u]);
    return p[u];
}
void join(int u,int v)
{
    int x=Find(u);
    int y=Find(v);
    if(x == y)
        return ;
    if(r[x]>r[y])
        p[y]=x;
    else if(r[x]<r[y])
        p[x]=y;
    else
    {
        p[x]=y;
        r[y]++;
    }
}
int kruskal()
{
    int ret;
    int cnt=0;
    for(int i=0;i<m;i++)
    {
        int u=e[i].u;
        int v=e[i].v;
        if(Find(u)!=Find(v))
        {
            cnt++;
            ret=e[i].w;
            join(u,v);
        }
        if(Find(1)==Find(n))
            break;
        if(cnt==n-1)
            break;
    }
    return ret;
}
int main()
{
    int T,count=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i = 1;i<=n;i++)
        {
            p[i]=i;
            r[i]=1;
        }
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        sort(e,e+m,cmp);
        int ans=kruskal();
        printf("Scenario #%d:\n%d\n\n",count++,ans);
    }
}