zzuli OJ 2271—魔法交流活动 51nod 1640（跑两次kruskal）

题目描述

魔法学校近日开展了主题为“天气晴朗”的魔法交流活动。
N名魔法师按阵法站好，之后选取N - 1条魔法链将所有魔法师的魔力连接起来，形成一个魔法阵。
魔法链是做法成功与否的关键。每一条魔法链都有一个魔力值V，魔法最终的效果取决于阵中所有魔法链的魔力值的和。
由于逆天改命的魔法过于暴力，所以我们要求阵中的魔法链的魔力值最大值尽可能的小，与此同时，魔力值之和要尽可能的大。
现在给定魔法师人数N，魔法链数目M。求此魔法阵的最大效果。

输入

两个正整数N，M。(1 <= N <= 10^5, N <= M <= 2 * 10^5) 
接下来M行，每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX) 
保证输入数据合法。

输出

输出一个正整数R，表示符合条件的魔法阵的魔力值之和。

样例输入

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

样例输出

12

首先跑一次最小生成树 记录下树里面的最大边。然后再对小于最大边的所有边跑一次最大生成树 即可。

#include<bits/stdc++.h>
using namespace std;

const int maxn=200000+5;
int u[maxn],v[maxn],w[maxn],p[maxn],r[maxn];

int cmp(int i,int j)
{
	return w[i]<w[j];
}
int find(int x)
{
	return p[x]==x?x:p[x]=find(p[x]);
}
int kruskal(int m)
{
	int Max_edge=-1;
	int ans=0;
	for(int i=0;i<m;i++) p[i]=i;
	for(int i=0;i<m;i++) r[i]=i;
	sort(r,r+m,cmp);
	for(int i=0;i<m;i++)
	{
		int e=r[i];
		int x=find(u[e]);
		int y=find(v[e]);
		if(x!=y)
		{
			ans+=w[e];
			Max_edge=max(Max_edge,w[e]);
			p[x]=y;
		}
	}
	return Max_edge;
}
long long Maxkruskal(int m,int M_edge)
{
	long long ans=0;
	for(int i=0;i<m;i++) p[i]=i;
	for(int i=0;i<m;i++) r[i]=i;
	sort(r,r+m,cmp);
	for(int i=m-1;i>=0;i--)
	{
		int e=r[i];
		int x=find(u[e]);
		int y=find(v[e]);
		if(x!=y&&w[e]<=M_edge)
		{
			ans+=w[e];
			p[x]=y;
		}
	}
	return ans;
}

int main()
{
	int n,m;cin>>n>>m;
	for(int i=0;i<m;i++) cin>>u[i]>>v[i]>>w[i];
	if(m==1) cout<<w[0]<<endl;
	else 
	{
		int M_edge=kruskal(m);
		memset(p,0,sizeof(p));memset(r,0,sizeof(r));
		cout<<Maxkruskal(m,M_edge)<<endl;
	}
	return 0;
}