【51nod1640】天气晴朗的魔法（最小生成树）

Description

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，表示符合条件的魔法阵的魔力值之和。

题解：

先用克鲁斯卡尔求出最小生成树的最大的边，然后在从小于等于这条边里求最大生成树。

代码：

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAX_N=2e5+100;
struct edge
{
    int u,v,cost;
};
bool cmp1(const edge &a,const edge &b)
{
    return a.cost<b.cost;
}
bool cmp2(const edge &a,const edge &b)
{
    return a.cost>b.cost;
}
edge a[MAX_N];
int n,m,maxx,k;
int par[MAX_N];
void init()
{
    for(int i=0;i<=n;i++)
        par[i]=i;
}
int find(int x)
{
    if(par[x]==x)
        return x;
    else
        return par[x]=find(par[x]);
}
void kr()
{
    sort(a,a+m,cmp1);
    init();
    int cns=0;
    for(int i=0;i<m;i++)
    {
        edge c=a[i];
        if(find(c.u)!=find(c.v))
        {
            par[find(c.u)]=find(c.v);
            if(maxx<c.cost)
            {
                maxx=c.cost;
                k=i;
            }
            cns++;
            if(cns>=n-1)
                break;
        }
    }
}
long long kr1()
{
    maxx=-INF;
    k=0;
    kr();         //求最小生成树的最大边
    sort(a,a+m,cmp2);
    init();
    int cns=0;
    long long res=0;
    for(int i=0;i<m;i++)
    {
        edge c=a[i];
        if(find(c.u)!=find(c.v)&&c.cost<=maxx)
        {
            par[find(c.u)]=find(c.v);
            res+=c.cost;
            cns++;
            if(cns>=n-1)
                break;
        }
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].cost);
    long long d=kr1();
    printf("%lld\n",d);
    return 0;
}