51nod 1640天气晴朗的魔法(克鲁斯卡尔，并查集）

1640 天气晴朗的魔法

题目来源： 原创

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题 收藏 关注
这样阴沉的天气持续下去，我们不免担心起他的健康。

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，表示符合条件的魔法阵的魔力值之和。
Input示例
4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6
Output示例

12

题解：首先简化逻辑，所求为显然为生成树，限制条件为：生成树的最大边尽可能的小，在此基础上，生成树的总权重尽可能大。显然在最小生成树的时候取得，最大边的最小值。（考虑克鲁斯卡尔算法的过程，如果此时的最大边不是最小的，那么就存在一个最大边小于当前最大边，也就是克鲁斯卡尔算法错误，不可嫩的）。找到最大边的最小值，那么显然从该值，倒序在进行一次克鲁斯卡尔算法，即求最大生成树。

注意点：

1.找到的是最大边的最小值，所以需要在往后找一找其他等于该边权重的边。

2.结果会爆int

代码：

#include<iostream>
#include<fstream>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<stdio.h>
#include<utility>
#include<algorithm>
#include<map>
#include<stack>
#include<set>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn = 2*1e5+5;
const int mod = 1e9+7;
const int INF = 1<<30;
const ll llINF = 1e18+999;

int n, m, pre[maxn], counter, max_index;
ll ans;
int Find(int x)
{
    return pre[x]==x? x : pre[x]=Find(pre[x]);
}
struct edge
{
    int a, b, w;
    friend bool operator< (edge e1, edge e2){
        return e1.w<e2.w;
    }
}in[maxn];
void init( )
{
    for(int i=0; i<maxn; i++)
        pre[i] = i;
    counter = 0;
    ans = 0;
}
int main( )
{
    //freopen("input.txt", "r", stdin);
    while(~scanf("%d%d", &n ,&m))
    {
        for(int i=0; i<m; i++)
            scanf("%d%d%d", &in[i].a, &in[i].b, &in[i].w);
        sort(in, in+m);
        init( );
        for(int i=0; i<m; i++)
        {
            int ra = Find(in[i].a), rb = Find(in[i].b);
            if(ra != rb)
            {
                pre[ra] = rb;
                counter++;
            }
            if(counter == n-1)
            {
                for(int j=i; j<m && in[j].w==in[i].w; j++)//往后找到所有等于此边权重的所有边
                    max_index = j;
                break;
            }
        }
        init( );
        for(int i=max_index; i>-1; i--)
        {
            int ra = Find(in[i].a), rb = Find(in[i].b);
            if(ra != rb)
            {
                pre[ra] = rb;
                ans += in[i].w;
                counter++;
            }
            if(counter == n-1)
                break;
        }
        cout<<ans<<endl;
    }
    return 0;
}