【图论】【Dijkstra】路径还原，求解最短路径

以Dijkstra算法为例，求解两个顶点之间的最短路径。

算法思想：

用prev[ j ]来记录最短路上顶点 j 的前驱。在d[ j ]被d[ j ] = d[ k ] + cost[ k ][ j ]更新时，修改prev[ j ]=k，这样就可以得到路径数组。这样在实现的时候、不断更新prev即可。

代码如下：

//基于Dijkstra算法的路径还原 
#include<iostream>
#include<vector>
#include<algorithm>
#define MAXN 55
#define INF 999999
using namespace std;

int prev[MAXN];
int V,E;
int cost[MAXN][MAXN];	
int d[MAXN];
bool used[MAXN];

//顶点s到其余顶点的最短路 
void dijkstra(int s)
{
	fill(d,d+V,INF);
	fill(used,used+V,false);
	fill(prev,prev+V,-1);
	d[s]=0;
	
	while(true)
	{
		int v=-1;
		for(int u=0;u<V;u++)
			if(!used[u]&&(v==-1||d[u]<d[v]))
				v=u;
		
		if(v==-1)
			break;
		used[v]=true;
		
		for(int u=0;u<V;u++)
		{
			if(d[u]>d[v]+cost[v][u])
			{
				d[u]=d[v]+cost[v][u];
				prev[u]=v;
			}
		}	
	}
}

//到顶点t的最短路还原 
vector<int> get_path(int t)
{
	vector<int> path;
	for(;t!=-1;t=prev[t])
	{
		path.push_back(t);		//不断沿着prev[t]走 直到t=s 
	}
	reverse(path.begin(),path.end());		//#include<algorithm>
	return path;
}

int main()
{
	
	vector<int> lu;
	
	for(int i=0;i<MAXN;i++)
		for(int j=0;j<MAXN;j++)				
			if(i==j)
				cost[i][j]=0;
			else
				cost[i][j]=INF;
	
	cin >> V >>E;
	int a,b,c;
	for(int i=0;i<E;i++)
	{
		cin >> a >> b >>c;
		cost[a][b]=c;
	}
	
	dijkstra(0);
	lu=get_path(3);
	
	for(int i=0;i<lu.size();i++)
		cout << lu.at(i) << " ";
	cout << endl;
	
	return 0;
}

/*
输入：
5  6
0 4 6
1 0 9
1 2 3
0 2 2
2 3 5
3 4 1
输出
0 2 3

*/