题解:即求多源最短路径问题,使用Floyd算法求解。
参考思路:弗洛伊德算法
一些细节:1、字符串处理方法:用一个string数组来存储所有地名,其结点编号为数组中相应下标 2、要考虑输入地名不存在和无解的情况
代码如下:
// Problem#: 1031
// Submission#: 5233116
// The source code is licensed under Creative Commons Attribution-NonCommercial-ShareAlike 3.0 Unported License
// URI: http://creativecommons.org/licenses/by-nc-sa/3.0/
// All Copyright reserved by Informatic Lab of Sun Yat-sen University
#include <iostream>
#include <string>
#include <queue>
using namespace std;
string node[1500]; // 储存地点
int n,cnt; // cnt是地点位置的个数
int road[1500][1500]; // 无向图
void search() // 核心部分:Floyd算法
{
for (int k = 0;k < cnt;k ++)
{
for (int i = 0;i < cnt;i ++)
{
for (int j = 0;j < cnt;j ++)
{
if (road[i][k] != -1 && road[k][j] != -1)
{
if (road[i][j] == -1)
road[i][j] = road[i][k] + road[k][j];
else
road[i][j] = road[i][k] + road[k][j] < road[i][j] ? road[i][k] + road[k][j] : road[i][j];
}
}
}
}
}
int find(string pos) // 动态建表
{
for (int i = 0;i < cnt;i ++)
{
if (node[i] == pos)
return i;
}
node[cnt] = pos;
cnt ++;
return cnt - 1;
}
int find2(string pos) // 最后输入两个地点时使用
{
for (int i = 0;i < cnt;i ++)
{
if (node[i] == pos)
return i;
}
return -1;
}
int main()
{
int c;
cin >> c;
while(c --)
{
for (int i = 0;i < 1500;i ++)
node[i] = "";
for (int i = 0;i < 1500;i ++) // 无向图初始化,-1代表两个位置无穷远或者说两点的距离还未计算过
{
for (int j = 0;j < 1500;j ++)
road[i][j] = -1;
}
for (int i = 0;i < 1500;i ++) // 对角线赋值为0
road[i][i] = 0;
cnt = 0;
string s,t;
int d;
cin >> n;
for (int i = 0;i < n;i ++) // 构建无向图
{
cin >> s >> t >> d;
road[find(s)][find(t)] = road[find(t)][find(s)] = d;
}
search();
cin >> s >> t;
int a = find2(s), b = find2(t);
if (a == -1 || b == -1) // 如果两位置不存在
{
if (s == t) // 若为相同的地名
cout << 0 << endl;
else
cout << -1 << endl; // 否则无解
}
else
cout << road[a][b] << endl;
}
}