题目链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1203
题意:求连通n个城市的最短路径。
——>>先把“任意”两点间的距离存到dist数组里,然后对其进行一次排序,接着开始对dist进行扫描,
如果扫描的两个点的树根相同,说明这两个城市已经相连,就不用加距离了,如果扫描到的两个点的树根不相同,
说明这两个城市还没有相连,sum要加上这个距离,最后输出sum即可。
带路径压缩:
#include <iostream>
#include <cmath>
#include <algorithm>
#include <iomanip>
using namespace std;
const int maxn = 100 + 10; //最多有100个城市
typedef struct Tnode //定义点类型
{
double x;
double y;
}node;
typedef struct Tdistance //定义距离类型,同时保存下是点map[n1]与点map[n2]之间的距离
{
int n1;
int n2;
double dis;
}cdis;
bool cmp(cdis d1, cdis d2) //定义排序方式,按间距从小到大的方式排
{
return d1.dis < d2.dis;
}
int fa[maxn]; //fa为结点的父亲,height为结点的高度
int find(int x) //返回结点x的根(是map中的编号)
{
return fa[x] == x ? x : (fa[x] = find(fa[x]));
}
bool judge(int x, int y) //判断、合并操作
{
int new_x = find(x);
int new_y = find(y);
if(new_x == new_y) return 1; //如果编号为x与编号为y的结点已在同一棵树中,返回1,说明这两个城市已相连
fa[new_y] = new_x;
return 0;
}
int main()
{
int N, i, j;
node map[maxn]; //输入的N个结点(N个城市的坐标)
cdis dist[maxn*maxn]; //用来存任意两点间的距离
int cnt = 1, first = 1; //cnt用来计数测试数据组数,first为输出标记
while(cin>>N)
{
if(N == 0) return 0; //当N为0时,返回
for(i = 0; i < N; i++) //输入N个点的坐标,存到map里
cin>>map[i].x>>map[i].y;
int m = 0; //m用来统计多少个两点间的距离(存map[i]与map[j]之间的距离,就不存map[j]与map[i]之间的距离)
for(i = 0; i < N; i++)
for(j = i+1; j < N; j++)
{
dist[m].n1 = i; //记录下结点map[i]的编号
dist[m].n2 = j; //记录下结点map[j]的编号
dist[m++].dis = sqrt(pow(map[i].x - map[j].x, 2) + pow(map[i].y - map[j].y, 2)); //计算两点间的距离
}
sort(dist, dist+m, cmp); //排序
for(i = 0; i < N; i++) fa[i] = i; //建N棵根树,初始化树根为其本身
double sum = 0; //连接各城市的总距离
for(i = 0; i < m; i++) //对m个距离进行扫描
{
int ok = judge(dist[i].n1, dist[i].n2); //判断这两个点是否在同一棵树中
if(!ok) //如果不在同一棵树中的话,把它们连在同一棵树中,距离加上
sum += dist[i].dis;
}
if(first) first = !first; //让第一组数据不输出空行,下面的测试数据输出空行
else cout<<endl;
cout<<"Case #"<<cnt++<<":"<<endl;
cout<<"The minimal distance is: "<<setiosflags(ios::fixed)<<setprecision(2)<<sum<<endl;
}
return 0;
}
容易理解小小的写法:
#include <iostream>
#include <cmath>
#include <algorithm>
#include <iomanip>
using namespace std;
const int maxn = 100 + 10; //最多有100个城市
typedef struct Tnode //定义点类型
{
double x;
double y;
}node;
typedef struct Tdistance //定义距离类型,同时保存下是点map[n1]与点map[n2]之间的距离
{
int n1;
int n2;
double dis;
}cdis;
bool cmp(cdis d1, cdis d2) //定义排序方式,按间距从小到大的方式排
{
return d1.dis < d2.dis;
}
int fa[maxn], height[maxn]; //fa为结点的父亲,height为结点的高度
int find(int x) //返回结点x的根(是map中的编号)
{
while(fa[x] != x)
{
x = fa[x];
}
return x;
}
bool judge(int x, int y) //判断、合并操作
{
int fx = find(x);
int fy = find(y);
if(fx == fy) //如果编号为x与编号为y的结点已在同一棵树中,返回1,说明这两个城市已相连
return 1;
else //否则
{
if(height[fx] > height[fy]) //当编号为x的结点所在树的高度>编号为y的结点所在树的高度时
fa[fy] = fx; //把fx设为fy的父亲
else if(height[fx] == height[fy]) //当编号为x的结点所在树的高度==编号为y的结点所在树的高度时
{
fa[fy] = fx; //把fx设为fy的父亲
height[fx]++; //同时把树fx的高度+1
}
else //当编号为x的结点所在树的高度<编号为y的结点所在树的高度时
fa[fx] = fy; //把fy设为fx的父亲
return 0; //返回0
}
}
int main()
{
int N, i, j;
node map[maxn]; //输入的N个结点(N个城市的坐标)
cdis dist[maxn*maxn]; //用来存任意两点间的距离
int cnt = 1, first = 1; //cnt用来计数测试数据组数,first为输出标记
while(cin>>N)
{
if(N == 0) return 0; //当N为0时,返回
for(i = 0; i < N; i++) //输入N个点的坐标,存到map里
cin>>map[i].x>>map[i].y;
int m = 0; //m用来统计多少个两点间的距离(存map[i]与map[j]之间的距离,就不存map[j]与map[i]之间的距离)
for(i = 0; i < N; i++)
for(j = i+1; j < N; j++)
{
dist[m].n1 = i; //记录下结点map[i]的编号
dist[m].n2 = j; //记录下结点map[j]的编号
dist[m++].dis = sqrt(pow(map[i].x - map[j].x, 2) + pow(map[i].y - map[j].y, 2)); //计算两点间的距离
}
sort(dist, dist+m, cmp); //排序
for(i = 0; i < N; i++) //建N棵根树
{
fa[i] = i; //初始化树根为其本身
height[i] = 1; //初始化树高为1(因为只有它自己)
}
double sum = 0; //连接各城市的总距离
for(i = 0; i < m; i++) //对m个距离进行扫描
{
int ok = judge(dist[i].n1, dist[i].n2); //判断这两个点是否在同一棵树中
if(!ok) //如果不在同一棵树中的话,把它们连在同一棵树中,距离加上
sum += dist[i].dis;
}
if(first) first = !first; //让第一组数据不输出空行,下面的测试数据输出空行
else cout<<endl;
cout<<"Case #"<<cnt++<<":"<<endl;
cout<<"The minimal distance is: "<<setiosflags(ios::fixed)<<setprecision(2)<<sum<<endl;
}
return 0;
}