概述:有一些城镇之间需要接通道路,然后给出城镇序号和每个城镇之间的距离,求在什么情况下城镇之间道路距离之和最小。
思路:很明显的最短树问题,类似于第一个问题,无非就是把邻接矩阵转换为每个城镇和城镇之间的距离,进行一部转换即可。
感想:看到这个问题就想到第一个问题,这样连核心代码都没换,直接套用,就是在最后main主函数时候出了问题,要将条件转换为邻接矩阵,需要定义map[y][x]=map[x][y],解决这个问题,即可出答案。
#include<iostream>
#include<string.h>
#include<fstream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
using namespace std;
const int INF = 1 << 20;
const int maxn = 105;
int sign[maxn];
int dis[maxn];
int n, m, sum;
int map[maxn][maxn];
void prim()
{
sign[1] = 1;
for (int i = 2;i <= n;i++)
{
dis[i] = map[1][i];
}
int now, min;
for (int i = 1;i < n;i++)
{
min = INF;
for (int j = 1;j <= n;j++)
if (!sign[j] && dis[j] < min)
{
min = dis[j];
now = j;
}
sum += min;
sign[now] = 1;
for (int j = 1;j <= n;j++)
if (!sign[j] && map[now][j] < dis[j])
dis[j] = map[now][j];
}
}
int main()
{
//ifstream cin("in.txt");
while (cin >> n && n)
{
int x, y;
sum = 0;
memset(sign, 0, sizeof(sign));
m = n*(n - 1) / 2;
for (int i = 1;i <= m;i++)
{
cin >> x >> y;
cin >> map[x][y];
map[y][x] = map[x][y];
}
prim();
cout << sum << endl;
}
return 0;
}