魔法跳舞链 51Nod - 1640

 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 <iostream>
#include <queue>
#include <string.h>
#include <stdio.h>
#include <math.h>
using namespace std;

const int MAX = 1000000;
int N,n;
int visit[MAX];

struct T
{
    int a,b;
    int len;
};

struct cmp
{
    bool operator() (T a,T b)
    {
        return a.len > b.len;
    }
};
struct cmp2
{
    bool operator() (T a,T b)
    {
        return a.len < b.len;
    }
};

int Find(int x)
{
    int r = x;
    while( r!= visit[r])
        r = visit[r];
    int l = x,y;
    while( visit[l] != r )
    {
        y = visit[l];
        visit[l] = r;
        l = y;
    }
    return r;
}

int mix(int x,int y)
{
    int TT = 0;
    int Tx = Find(x);
    int Ty = Find(y);
    if(Tx!=Ty)
    {
        visit[Tx] = Ty;
        TT = 1;
    }
    return TT;

}

priority_queue<T,vector<T>,cmp>P1;
priority_queue<T,vector<T>,cmp2>P2;


int main()
{
    cin>>N>>n;
    for(int i =0;i<=N;i++)
        visit[i] = i;
    T temp ;
    for(int i =1;i<=n;i++)
    {
        cin>>temp.a>>temp.b>>temp.len;
        P1.push(temp);
        P2.push(temp);
    }

    int tn = 0;
    int max1 = 0;

    while(!P1.empty())
    {
        temp = P1.top();
        P1.pop();
        if(mix(temp.a,temp.b))
        {
            tn++;
            max1 = max(max1,temp.len);
        }
    }

    for(int i =0;i<=N;i++)
        visit[i] = i;
    long long sum = 0;
    while(!P2.empty())
    {
        temp = P2.top();
        P2.pop();
        if(temp.len<=max1)
            if( mix(temp.a,temp.b))
            {
                sum+=temp.len;
            }
    }
    cout<<sum<<endl;

    return 0;
}