图论 路径还原

个人模板

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int max_v = 1e3+5;
const int INF = 0x3f3f3f3f;

int prev[max_v];//最短路上的前趋顶点
int d[max_v];
int cost[max_v][max_v];
int used[max_v];

int V, E;

//求从起点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;//v点已经取过 
		
		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());//这样得到的是按照t到s的顺序，所以翻转之
	return path; 
}

int main()
{
	scanf("%d %d", &V, &E);
	for(int i = 0; i < E; i++)
	{
		int u, v, w;
		scanf("%d %d %d", &u, &v, &w);
		cost[u][v] = cost[v][u] = w;
	}
	
	int s, t;//起点 和 终点 
	scanf("%d", &s, &t); 
	
	dijkstra(s);
	vector<int> vec = get_path(t);
	
	vector<int>::iterator p;
	for(p = vec.begin(); p != vec.end(); ++p) 
	{
		printf("%d ", *p);
	}
	printf("\n");
	
	return 0;
}