POJ 3522 kruskal 枚举 贪心

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxv=110;
const int maxe=9000;
const int INF=0x3f3f3f3f;
int pre[maxv],cnt;
struct Edge {
    int from,to,val;
    bool operator<(const Edge& rhs)const{
        return val<rhs.val;
    }
} edge[maxe];
void init() {
    cnt=0;
}
void addedge(int from,int to,int val) {
    edge[cnt].from=from;
    edge[cnt].to=to;
    edge[cnt++].val=val;
}
int Find(int x){//找到根节点
    int r=x;
    while(r!=pre[r])
        r=pre[r];
    int i=x,j;
    while(pre[i]!=r){
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
int Kruskal(int n) { //n代表图种总共的顶点数
    int mmin,mmax,ecnt=0,ans=INF;
    sort(edge,edge+cnt);
    for(int i=0; i<=(cnt-n+1); i++) {
//贪心思想，Kruskal算法是最小生成树算法，所以找到权值
//的最大值一定是最小的，我们现在要让最小值尽可能大，
//所以我们先sort，然后从权值最小的开始遍历，找生成树
//每次抛弃一个最小的权值，那么上限是多少呢?一共有m条边
//树需要有n-1条边，所以上限是m-(n-1)
        ecnt=0;
        for(int i=0;i<=n;i++)
            pre[i]=i;
        mmin=INF,mmax=-1;
        for(int j=i; j<cnt; j++) {
            int f1=Find(edge[j].from);
            int f2=Find(edge[j].to);
            if(f1!=f2) {
                pre[f1]=f2;
                if(mmin>edge[j].val)
                    mmin=edge[j].val;
                if(mmax<edge[j].val)
                    mmax=edge[j].val;
                ecnt++;
                if(ecnt==n-1)
                    break;
            }
        }
        if(ecnt==n-1) {
            ans=min(ans,mmax-mmin);
        }
    }
    return ans==INF?-1:ans;//返回-1表示无法生成最小生成树
}
int main(void) {
#ifndef ONLINE_JUDGE
    freopen("E:\\input.txt","r",stdin);
#endif
    ios::sync_with_stdio(false);
    int n,m,t1,t2,t3;
    while(cin>>n>>m&&n+m) {
        init();
        for(int i=0; i<m; i++) {
            cin>>t1>>t2>>t3;
            addedge(t1,t2,t3);
        }
        cout<<Kruskal(n)<<endl;
    }
    return 0;
}