Bad Cowtractors POJ 2377【克鲁斯卡尔算法】

G - Bad Cowtractors

 POJ - 2377 

题意：求一个最大生成树，如果这个图是森林，输出 -1，不是的话输出权值

分析：最大生成树与最小生成树是反着的，只需要将cmp函数的方向改下，判断这个图是不是森林，只需要每次合并时将count++即可，最后判count是否等于n-1即可

#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=1e3+50;
const int MAXM=2e4+50;

struct edge{
    int u,v,w;
};
edge edges[MAXM];

bool cmp(edge a,edge b)
{
    return a.w>b.w;
}

int f[MAXN];
//并查集

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

void merge(int a,int b)
{
    int t1=find(a);
    int t2=find(b);
    if(t1!=t2){
        f[t2]=t1;
    }
    return ;
}
//克鲁斯 卡尔算法
void krusal(int n,int m)
{
    sort(edges,edges+m,cmp);//先排序
    int ans=0,count=0;
    for(int i=0;i<m;i++){
        int u=find(edges[i].u);
        int v=find(edges[i].v);
        if(u!=v){//不连通加上此时的权值
            merge(u,v);
            ans+=edges[i].w;
            count++;
        }else{//否则跳过即可
            continue;
        }
    }
    if(count!=n-1)
       printf("-1");
    else
        printf("%d",ans);



}
int main(){
    for(int i=0;i<=MAXN;i++){
            f[i]=i;
    }
   int n,m;
   scanf("%d%d",&n,&m);

   for(int i=0;i<m;i++){
       scanf("%d%d%d",&edges[i].u,&edges[i].v,&edges[i].w);
   }
   krusal(n,m);

    return 0;
}