SSL2817 2017年11月1日提高组 树论（并查集）

2017年11月1日提高组 树论

Description

Input

Sample Input

1
4 2
3 1 3375
3 2 8707
5 5
2 1 455
3 1 3616
4 1 9494
5 4 1448
5 1 8366
Sample Output

-1
7911
Hint

分析：按边权从小到大排序，枚举最小边，往后把边加入并查集，如果n个点都属于同一集合时就统计答案

代码

#include <cstdio>
#include <algorithm>
#define N 5000
using namespace std;

struct arr
{
    int x,y,w;
}a[N];
int f[N],n,m;

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

int min(int x,int y)
{
    return x<y?x:y;
}

int so(arr p,arr q)
{
    return p.w<q.w;
}

int main()
{
//  freopen("graph.in","r",stdin);
//  freopen("graph.out","w",stdout);
    int T;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&n,&m);
        for (int i=1;i<=m;i++)
            scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].w);
        sort(a+1,a+m+1,so);
        int ans=123456789;
        for (int i=1;i<=m;i++)
        {
            int tot=0;
            for (int j=1;j<=n;j++)
                f[j]=j;
            int j=i;
            while (j<=m)
            {
                int u=find(a[j].x);
                int v=find(a[j].y);
                if (u!=v)
                {
                    tot++;
                    f[u]=v;
                    if (tot==n-1) break;
                }
                j++;
            }
            if (tot==n-1) ans=min(ans,a[j].w-a[i].w);
        }
        if (ans==123456789) ans=-1;
        printf("%d\n",ans);
    }
    fclose(stdin);
    fclose(stdout);
}