试题编号: | 201703-4 |
试题名称: | 地铁修建 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 A市有n个交通枢纽,其中1号和n号非常重要,为了加强运输能力,A市决定在1号到n号枢纽间修建一条地铁。 输入格式 输入的第一行包含两个整数n, m,用一个空格分隔,分别表示交通枢纽的数量和候选隧道的数量。 输出格式 输出一个整数,修建整条地铁线路最少需要的天数。 样例输入 6 6 样例输出 6 样例说明 可以修建的线路有两种。 评测用例规模与约定 对于20%的评测用例,1 ≤ n ≤ 10,1 ≤ m ≤ 20; |
问题连接:CCF201703-4 地铁修建
解题思路:题目要求把枢纽1和枢纽n连通的方案中最大的边最小,把边从小到大排列,每加一条边(加入这条边后不会形成回路)就判断枢纽1和枢纽n是否连通,如果连通这就是答案,程序使用并查集判断枢纽1和枢纽n是否连通
提交后得100分的C++程序:
#include<iostream>
#include<algorithm>
using namespace std;
const int N=200005;
//并查集
int pre[N];
void init(int n)
{
for(int i=0;i<=n;i++)
pre[i]=i;
}
int find(int x)
{
int r=x;
while(r!=pre[r])
r=pre[r];
//路径压缩
while(x!=pre[x]){
int i=pre[x];
pre[x]=r;
x=i;
}
return r;
}
bool join(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy){
pre[fx]=fy;
}
}
struct Edge{//边
int u,v,w;
bool operator<(const Edge &a)const
{
return w<a.w;
}
}a[N];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++)
scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
sort(a,a+m);
init(n);
for(int i=0;i<m;i++){
join(a[i].u,a[i].v);//添加这条边
if(find(1)==find(n)){//如果添加这条边后1和n连通了输出w后退出
printf("%d\n",a[i].w);
break;
}
}
return 0;
}