week6_D 数据中心（kruskal最小生成树）

题目概述

思路分析

上课的时候这道题目读完第一遍一头雾水，学长引导了一下才把握题意。

本道题的任务是：每个节点需要选择一条路径将数据发送到root节点。

关键的一句话：root节点只能接收数据，其余任何一个节点可以将数据传输给另外一个节点，但是不能将数据传输给多个节点。
不能将数据传输给多个节点意味着每个节点到root节点的路径只能有一条，我们的任务就是找出所有可能情况中耗时最短的一条路线。那么就可以用kruskal最小生成树解决，找到最大边的权值最小的那条路径。

实现：用sort将边按照权值由小到大进行排序，每次贪心的取权值最小边，用并查集判断若不构成环路，则加入结果中，直到结果中有n-1条边时结束。

代码

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;

int n,m,root,tot=1;
const int maxn=5e4+5;
int par[maxn];
struct Edge{
 int u,v,w;
 bool operator<(const Edge& e){
  return w<e.w;
 }
}edge[100005];

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

bool unite(int x,int y)
{
 x=find(x);
 y=find(y);
 if(x==y) return false;
 par[x]=y;
 return true;
}

int kruskal()
{
 for(int i=1;i<=n;i++) par[i]=i;
 sort(edge,edge+m+1);
 int ans=0,num=0;
 for(int i=1;i<=m;i++)
 {
  if(unite(edge[i].u,edge[i].v))
  {//两个端点没连通
   ans=max(ans,edge[i].w);//找到权值最大边
   if(++num==n-1) break;
  }
  } 
 return num==n-1?ans:-1;
}

int main()
{
 cin>>n>>m>>root;
 int u,v,w;
 for(int i=1;i<=m;i++)
 {
  cin>>edge[tot].u>>edge[tot].v>>edge[tot].w;
  tot++;
 }
 cout<<kruskal()<<endl;
 return 0;
}