路径还原

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#define INF 999999
using namespace std;

int G[203][203];
int n, s, t;

bool Dijkstra()
{
    int i, j, k;
    int dis[203];
    bool vis[203];
    int pre[203];//记录路径

    memset(vis,false,sizeof(vis));
    fill(dis,dis+n,INF);
    memset(pre,-1,sizeof(pre));//初始化为-1
    dis[s] = 0;
    for(i=0;i<n;i++)
    {
        k = -1;
        for(j=0;j<n;j++)
            if(vis[j] == false && (k == -1 || dis[k]>dis[j]))
                k = j;
        if(k == -1)
            break;
        vis[k] = true;
        for(j=0;j<n;j++)
            if(vis[j] == false && dis[j]>dis[k]+G[k][j]){
                dis[j] = dis[k]+G[k][j];
                pre[j] = k;//在松弛操作中更新边的同时  记录路径
            }
    }
    printf("%d——>%d  的最短路为 ：",s,t);
    printf("%d\n",dis[t]);
    printf("路径为 ：");
    //一下部分为路径的还原
    stack<int >que;
    for(t;t!=-1;t=pre[t])//从t 一直寻找到s
        que.push(t);//放入栈中
    printf("%d",que.top());//输出第一个
    que.pop();//删除
    while(!que.empty()){
        printf("——>%d",que.top());//输出
        que.pop();//删除
    }
    printf("\n");
}

int main()
{
    int i, j, u, v, w, m;

    while(scanf("%d %d",&n,&m)==2){
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                G[i][j] = i==j?0:INF;
        while(m--){
            scanf("%d %d %d",&u,&v,&w);
            if(G[u][v]>w)
                G[u][v] = G[v][u] = w;
        }
        scanf("%d %d",&s,&t);
        Dijkstra();
    }
    return 0;
}
/*
3 3
0 1 1
0 2 3
1 2 1
0 2

2
0->1->2
*/