畅通工程
Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 13854 Accepted Submission(s): 5727
Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
Output
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
Sample Input
3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100
Sample Output
3 ?
kruskal算法思想:
在Kruskal算法中判断加入边是否形成‘回路’(注意,不是环路)的方法可以通过类似集合的运算思想。一开始n个顶点分别各自属于一个集合(标号1-n),每加入一条边相当于完成了两个集合的合并。比如加入边<i,j>就是把j顶点集合中元素标号改为与i顶点所在集合元素标号相同。
当需要加入的i,j都是同一个集合内时,则舍去该边(构成回路)。
完成n-1次合并则算法结束。
算法框架:
T=(V,φ)while(T中所含边数<n-1)
{
从E中选取当前最短边(u,v);
从E中删去边(u,v);
if( (u,v)加入T中后不产生回路)
将边(u,v)加入T中;
}
如何判断是否出现回路?
初始时将每个点置入不同的集合,向生成树中加边时,判断该边所连两点是否属于同一集合。若不在同一集合,则将此边加入生成树,并合并此两点所在集合;否则不加此边。直到加够n-1条边为止。
一般使用并查集
设G=(V,E)是网络,最小生成树的初始状态为只有n个顶点而无边的非连通图T=(V,φ),T中每个顶点自成为一个连通分量。
将集合E中的边按权递增顺序排列,从小到大依次选择顶点分别在两个不同连通分量中的边加入图T,则原来的两个连通分量由于该边的连接而成为一个连通分量。
依次类推,直到T中所有顶点都在同一个连通分量上为止,该连通分量就是G的一棵最小生成树。
解题思路:
最小生成树,kruskal算法,初始化时把每点之间的权值设为无穷大,后面合并加入了最小生成树的点,使用并查集完成,最后判断的时候只要
cost 大于 设置的无穷大 maxint 就说明村庄没有连通,输出 ? ,其他输出cost。
代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int LEN=105;
const int maxint=0x3f3f3f3f;
int path[LEN][LEN], f[LEN], dis[LEN];
struct node
{
int st, ed, w;
} road[LEN*LEN];
int find_fa(const int& a)
{
return f[a]==a ? a : f[a]=find_fa(f[a]);//查找根节点同时更新子节点保存的根节点下标
}
//可以不用
//void unite(const int &x, const int &y)
//{
// if( x==y )
// return ;
// // if( dis[x]>dis[y] ) //这里的目的是使得连上以后的树不要太深造成查找超时
// f[y]=x;
// // else
// // {
// // f[x]=y;
// // if( dis[x]==dis[y] )
// // dis[y]++;
// // }
//}
int cmp(const void *a, const void *b)
{
node x=*(node *)a;
node y=*(node *)b;
return x.w-y.w;
}
int kruskal(const int& n)
{
int cost=0, i, j, k=0;
for(i=1; i<=n; i++)
for(j=1; j<=n; j++)
{
road[k].st=i;
road[k].ed=j;
road[k++].w=path[i][j];
}
qsort(road, k, sizeof(node), cmp);
for(i=1; i<=n; i++)
{
f[i]=i;//并查集初始化
// dis[i]=0;
}
for(k=j=0; j<n-1; )
{
int x=find_fa(road[k].st);//并查集状态压缩过,所以查找的时间会大大减少
int y=find_fa(road[k].ed);
if( x!=y )//如果不在同一个集合,则合并,否则舍去这条边
{
j++;
cost+=road[k].w;
f[y]=x;
//unite(x, y);
}
k++;
}
return cost;
}
int main()
{
int n, m, i, j, Q, a, b, cost=0;
while( cin>>n>>m && n )
{
cost=0;
for(i=1; i<=m; i++)
for(j=1; j<=m; j++)
{
path[i][j]=path[j][i]=maxint;
if( i==j ) path[i][j]=0;
}
for(i=1; i<=n; i++)
scanf("%d%d%d", &a, &b, &Q),
path[a][b]=path[b][a]=Q;
cost=kruskal(m);
if( cost>=maxint )
cout<<"?\n";
else cout<<cost<<endl;
}
return 0;
}