Description
n个点,边有边权,点有点权,定义两点之间路径长度为路径上边权和加上路径上除起点终点外的点权和,给出多组查询,每次给出起点和终点,问两点间最短距离
Input
多组用例,每组用例第一行输入一整数n表示点数,之后一个n*n矩阵A表示邻接矩阵,然后输入n个值val[i]表示第i个点的点权,之后输入多组查询,每组查询输入两个整数s,e表示查询s到e的最短距离,以-1 -1结束查询,以n=0结束输入
Output
对于每组用例,输出s到e的最短距离以及对应的路径,如果有多种合法路径输出字典序最小的那个,每组输出后跟一空行
Sample Input
5
0 3 22 -1 4
3 0 5 -1 -1
22 5 0 9 20
-1 -1 9 0 4
4 -1 20 4 0
5 17 8 3 1
1 3
3 5
2 4
-1 -1
0
Sample Output
From 1 to 3 :
Path: 1–>5–>4–>3
Total cost : 21
From 3 to 5 :
Path: 3–>4–>5
Total cost : 16
From 2 to 4 :
Path: 2–>1–>5–>4
Total cost : 17
Solution
Floyd,加一个Next[i][j]记录i到j的最短路径中i的直接后继,更新的时候如果a[i][j]>a[i][k]+a[k][j]+b[k]或者a[i][j]==a[i][k]+a[k][j]+b[k]&&Next[i][j]>Next[i][k],就更新Next[i][j]为Next[i][k]
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 1111
int n,a[maxn][maxn],b[maxn],Next[maxn][maxn];
int main()
{
while(~scanf("%d",&n),n)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&a[i][j]);
if(a[i][j]==-1)a[i][j]=INF;
Next[i][j]=j;
}
for(int i=1;i<=n;i++)scanf("%d",&b[i]);
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if(a[i][j]>a[i][k]+a[k][j]+b[k])
{
a[i][j]=a[i][k]+a[k][j]+b[k];
Next[i][j]=Next[i][k];
}
else if(a[i][j]==a[i][k]+a[k][j]+b[k]&&Next[i][j]>Next[i][k])
Next[i][j]=Next[i][k];
}
int u,v;
while(~scanf("%d%d",&u,&v),~u||~v)
{
int c=a[u][v];
printf("From %d to %d :\n",u,v);
printf("Path: %d",u);
while(u!=v)
{
printf("-->%d",Next[u][v]);
u=Next[u][v];
}
printf("\nTotal cost : %d\n\n",c);
}
}
return 0;
}