蓝桥公园(最短路径:Floyd算法)
题目描述:
小明喜欢观景,于是今天他来到了蓝桥公园。已知公园有 N 个景点,景点和景点之间一共有 M 条道路。小明有 Q 个观景计划,每个计划包含一个起点 st 和一个终点ed,表示他想从 st 去到 ed。但是小明的体力有限,对于每个计划他想走最少的路完成,你可以帮帮他吗?
输入描述:
输入第一行包含三个正整数 N,M,Q第 2 到 M + 1 行每行包含三个正整数 u,v,w,表示 u <–> v 之间存在一条距离为 w 的路。
第 M+2 到 M + Q-1 行每行包含两个正整数 st,ed,其含义如题所述
1 <=N <=400,1<= M <= (N*(N-1)/2),Q <= 103,1 <= u,v,st,ed <=n,1 <= w <= 109
输出描述:
输出共 Q 行,对应输入数据中的查询。若无法从st到达ed则输出-1。
输入样例:
3 3 3
1 2 1
1 3 5
2 3 2
1 2
1 3
2 3
输出样例:
1
3
2
思路:
这道题主要运用最短路径中的Floyd算法来解决,算法只有5行,比较简短。需要注意的是,在输入路径及权值时,一条路有可能重复输入,要选取其中最短的。查询的时候,如果查询的路径在数组中还是INF,也就是没改变,则没有最短路径,输出-1,如果查询的路径是到它本身,则路径为0,否则输出最短路径。
要注意无穷大的定义。
代码:
#include<bits/stdc++.h>
using namespace std;
long long INF=0x3f3f3f3f3f3f3f3f; //定义无穷大变量
int n,m,t;
long long dp[500][500]; //存储最短路
void input()
{ //输入函数
int x,y;
long long z;
scanf("%d %d %d",&n,&m,&t);
for(int i=0;i<m;i++)
{
scanf("%d %d %lld",&x,&y,&z);
dp[x][y]=dp[y][x]=min(dp[x][y],z); //防止路径重复
}
}
void Floyd()
{ //最短路径函数(Floyd算法)
for(int k=1;k<=n;k++)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
}
}
void ouput()
{ //输出函数
int v,u;
while(t--)
{
scanf("%d %d",&v,&u);
if(dp[v][u]==INF) printf("-1\n");
else if(v==u) printf("0\n");
else printf("%lld\n",dp[v][u]);
}
}
int main()
{
for(int i=0;i<=500;i++)
{ //初始化为无穷大
for(int k=0;k<=500;k++)
dp[i][k]=INF;
}
input();
Floyd();
ouput();
return 0;
}