程序设计Week6——D-数据中心

题目描述

有一个n个节点的网络图，编号从1到n，全双工（无向图），两点相连代表可以发送数据，边权为发送数据消耗时间。现在每一个节点都需要发送数据到root号节点，求一个最优树结构，使得该任务完成时间最少。
PS：root点只能接收数据，其余节点可以发送和接收数据，发送时只能传输给另一个节点。
一个树结构图的传输时间Tmax=max（Th），h为接收点在树中的深度，Th=max（Th,j），Th,j表示j条不同的边，所有边的深度均为h。

Input

第一行一个数n，1<=n<=5e4
第二行一个数m，1<=m<=1e5
第三行一个数root，1<=root<=5e4
接下来m行，每行三个数u，v，w，代表每条边连接的顶点，和该边边权

Output

一个正整数ans，表示最优树结构的耗时

解题思路

首先根据题目，将其化为显式的图结构，题目给出一个带权无向图，求解的是一个最大边权最小的生成树，换而言之，就是求一个最小生成树，但现在判定标准不再是边权总和最小，而是最大边权最小。因此依然可以使用Kruskal算法，只需要在加边时，将计算sum（边权和），更换为计算ans（当前边集内最大边权值），其余基本一致，最后输出ans即可。

实现代码

#include<iostream>
#include<algorithm>
using namespace std;

const int maxn=100005;
int f[maxn];
struct edge
{
	int u,v,w;
	bool operator<(const edge &p) const
	{return w<p.w;}
};
edge e[maxn];

int find(int x)
{
	if(f[x]!=x)
		f[x]=find(f[x]);
	return f[x];
}

bool combine(int u,int v)
{
	int x=find(u),y=find(v);
	if(x==y)
		return false;
	f[x]=y;
	return true;
}

int main()
{
	int n,m,root;
	cin>>n>>m>>root;
	for(int i=1;i<=m;++i)
		cin>>e[i].u>>e[i].v>>e[i].w;
	for(int i=1;i<=n;++i)
		f[i]=i;
	sort(e+1,e+1+m);
	int total=0,ans=-1;
	for(int i=1;i<=m;++i)
	{
		if(combine(e[i].u,e[i].v))
		{
			ans=max(ans,e[i].w);
			total++;
			if(total==n-1)
				break;
		}
	}
	cout<<ans<<endl;
	return 0;
}

总结

这道题与C是类似的，核心都是使用Kruskal算法，并且都需要根据题目信息转化成显式的图求解思路，区别仅在于C题求的是标准的最小生成树，而本题是求解最大边权最小的最小生成树，仅需在加入边时加以修改即可。