天气晴朗的魔法 (Kruskal)

这样阴沉的天气持续下去，我们不免担心起他的健康。

51nod魔法学校近日开展了主题为“天气晴朗”的魔法交流活动。

N名魔法师按阵法站好，之后选取N - 1条魔法链将所有魔法师的魔力连接起来，形成一个魔法阵。

魔法链是做法成功与否的关键。每一条魔法链都有一个魔力值V，魔法最终的效果取决于阵中所有魔法链的魔力值的和。

由于逆天改命的魔法过于暴力，所以我们要求阵中的魔法链的魔力值最大值尽可能的小，与此同时，魔力值之和要尽可能的大。

现在给定魔法师人数N，魔法链数目M。求此魔法阵的最大效果。

Input

两个正整数N，M。(1 <= N <= 10^5, N <= M <= 2 * 10^5)

接下来M行，每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX)

保证输入数据合法。

Output

输出一个正整数R，表示符合条件的魔法阵的魔力值之和。

Sample Input

4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6

Sample Output

12

思路：

先找到一棵最小生成树，记录权值最大的边

在不超过最大权值的边中，组成一棵最大生成树

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAX 200005
using namespace std;
struct node{
    int u,v;
    int w;
}num[MAX];
int pre[MAX];
int n,m;
long long  sum1,sum2;
int maxx;
void init(){  //初始化
    int i;
    for(i=0;i<=n;i++)
        pre[i]=i;
}
int Find(int x){
    int r=x;
    while(r!=pre[r])  //寻找根结点
        r=pre[r];
    int i=x,j;
    while(i!=r){     //路径压缩
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
int mix(int a,int b){
    int fa=Find(a),fb=Find(b);
    if(fa!=fb){
        pre[fb]=fa;
        return 1;
    }
    return 0;
}
int cmp(node a,node b){
    return a.w<b.w;
}
void Krusal(int x);
int main(){
    scanf("%d%d",&n,&m);
    int i;
    for(i=0;i<m;i++){
        scanf("%d%d%d",&num[i].u,&num[i].v,&num[i].w);
    }
    sort(num,num+m,cmp);//按权值从小到大排序
    Krusal(1); //找最小生成树
    Krusal(2); //在不超过最大权值的边中，找到一棵最大生成树
    long long sum=sum1>sum2?sum1:sum2;
    printf("%lld\n",sum);
return 0;
}
void Krusal(int x){
    init();
    int cnt=0;
    int i;
    if(x==1){  
        for(i=0;i<m;i++){
            if(mix(num[i].u,num[i].v)){
                sum1+=num[i].w;;
                cnt++;
            }
            if(cnt==n-1) break;
        }
        maxx=num[i].w; //最大的边一定是加入树的最后一条
    }
    else{
        for(i=m-1;i>=0;i--){
            if(num[i].w<=maxx&&mix(num[i].u,num[i].v)){
                sum2+=num[i].w;;
                cnt++;
            }
            if(cnt==n-1) break;
        }
    }
}