这样阴沉的天气持续下去,我们不免担心起他的健康。
51nod魔法学校近日开展了主题为“天气晴朗”的魔法交流活动。
N名魔法师按阵法站好,之后选取N - 1条魔法链将所有魔法师的魔力连接起来,形成一个魔法阵。
魔法链是做法成功与否的关键。每一条魔法链都有一个魔力值V,魔法最终的效果取决于阵中所有魔法链的魔力值的和。
由于逆天改命的魔法过于暴力,所以我们要求阵中的魔法链的魔力值最大值尽可能的小,与此同时,魔力值之和要尽可能的大。
现在给定魔法师人数N,魔法链数目M。求此魔法阵的最大效果。
接下来M行,每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX)
保证输入数据合法。
4 6 1 2 3 1 3 1 1 4 7 2 3 4 2 4 5 3 4 6
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;
}
}
}