天气晴朗的魔法

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

Input示例

4 6
1 2 3
1 3 1
1 4 7
2 3 4
2 4 5
3 4 6

Output示例

12

先求一遍最小生成树，最小生成树中的最大边权为上界（因为要满足魔力值的最大值尽量小），然后以此上界为标准求一次最大生成树（使得魔力值总和尽量大）

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int pre[200005];
struct list
{
	int a;
	int b;
	long long val;
}an[200005];
bool cmp(struct list a,struct list b)
{
	return a.val<b.val;
}
int find(int a)
{
	return pre[a]==a?a:pre[a]=find(pre[a]);
}
int main()
{
	int n,m,cnt=0;
	long long sum=0,upper=0;
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++)
	{
		scanf("%d%d%lld",&an[i].a,&an[i].b,&an[i].val);
	}	
	sort(an,an+m,cmp);
	for(int i=1;i<=n;i++)
		pre[i]=i;
	for(int i=0;i<m;i++)
	{
		int u,v;
		u=find(an[i].a);
		v=find(an[i].b);
		if(u!=v)
		{
			pre[u]=v;
			upper=max(upper,an[i].val);
			if(++cnt==n-1)
				break;
		}
	}
	cnt=0;
	for(int i=1;i<=n;i++)
		pre[i]=i;
	for(int i=m-1;i>=0;i--)
	{
		int u,v;
		u=find(an[i].a);
		v=find(an[i].b);
		if(an[i].val<=upper&&u!=v)
		{
			pre[u]=v;
			sum+=an[i].val;
			if(++cnt==n-1)
				break;
		}
	}
	printf("%lld",sum);
	return 0;
}