POJ 3522 Slim Span

题目链接http://poj.org/problem?id=3522

kruskal+并查集，注意特殊情况比如1,0 、0,1、1,1

#include<cstdio>
#include<iostream>
#include<algorithm> 
#include<climits>
using namespace std;
#define MAXN 5005
struct edge{
    int u,v,cost;
};
int comp(const edge& e1,const edge& e2){
    return e1.cost<e2.cost;
}
edge es[MAXN];
int V,E;
int par[MAXN],rank[MAXN];
void init(){
    for(int i=1;i<=V;i++){
    par[i]=i;
    }
} 
int find(int x){
if(par[x]==x)return x;
return par[x]=find(par[x]);    
}
void unite(int x,int y){
    x=find(x);
    y=find(y);
    if(x!=y){
    par[x]=y;
    }
}
bool same(int x,int y){
    return find(x)==find(y);
}
int kruskal(int k){
    init();
    int min,max=-1;
    int esum=0;
    for(int i=k;i<E;i++){
        if(E-k<V-1)break;
        edge e=es[i];
        if(!same(e.u,e.v)){
            unite(e.u,e.v);
            esum++;
            max=max<e.cost?e.cost:max;
        }
        if(esum==V-1){
            min=es[k].cost;
            return max-min;
        }
    }
    return -1;
}
int main()
{
    int i,flag=0;
    while(scanf("%d%d",&V,&E)==2){
    if(V==0&&E==0)break;
    for(i=0;i<E;i++){
    scanf("%d%d%d",&es[i].u,&es[i].v,&es[i].cost);
    }
    if(E==0||V>100||E<V-1){
    printf("-1\n");continue;}
    sort(es,es+E,comp);
    int min=INT_MAX,x;
    for(i=0;i<E-V+2;i++){
       x=kruskal(i);
       if(min>x&&x!=-1){
       min=x;
       }
    }
    if(min!=INT_MAX)
    printf("%d\n",min);
    else
    printf("-1\n");
    }
    return 0;
}