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;
}