1. 程式人生 > >Dijkstra求最短路的條數,並輸出最短路徑和最短路經過的點的最大和

Dijkstra求最短路的條數,並輸出最短路徑和最短路經過的點的最大和

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <stack>


using namespace std;


const int maxn=1e3;
const int INF=1e8;
int n,m,st,en,x,y,z;
int val[maxn],valsum[maxn],path[maxn],pathsum[maxn];//val儲存每個點的值,valsum儲存到點【i】前的點值和,path儲存點【i】前那個點的號,pathsum儲存到點【i】的最短路的條數
int d[maxn],vis[maxn],G[maxn][maxn];


void dijkstra()
{
    for(int i=0;i<n;i++)
        d[i]=G[st][i];
    d[st]=0;
    pathsum[st]=1;//初始化起點的最短路的條數是1
    int min1,p;
    for(int i=1; i<=n; i++)
    {
        min1=INF;
        for(int j=0; j<n; j++)
        {
            if(!vis[j] && d[j]<min1)
            {
                min1=d[j];
                p=j;
            }
        }
        vis[p]=1;
        for(int j=0; j<n; j++)
        {
            if(!vis[j] && d[j]>d[p]+G[p][j])
            {
                d[j]=d[p]+G[p][j];
                pathsum[j]=pathsum[p];//當鬆弛時,到【j】和到【p】的最短路的條數相同
                valsum[j]=valsum[p]+val[j];//鬆弛時直接加上更新後的點值
                path[j]=p;//記錄上一個點
            }
            else if(!vis[j] && d[j]==d[p]+G[p][j])
            {
                pathsum[j]+=pathsum[p];//當相等時,到【j】的加上到【p】的
                if(valsum[j]<valsum[p]+val[j])
                {
                    valsum[j]=valsum[p]+val[j];//當相等時加上大的那個
                    path[j]=p;//相等時記錄大的那個點
                }
            }
        }
    }
}


void P()//輸出路徑
{
    int tmp;
    stack<int>si;
    while(!si.empty()) si.pop();
    si.push(en);//從最後的那個點開始逆序存到堆疊裡
    while (st!=en)
    {
        tmp=path[en];
        si.push(tmp);
        en=tmp;
    }
    cout << si.top();
    si.pop();
    while(!si.empty())
    {
        cout << ' ' << si.top();
        si.pop();
    }
    cout << endl;
}


void inition()
{
    memset(path,0,sizeof(path));
    memset(vis,0,sizeof(vis));
    memset(pathsum,0,sizeof(pathsum));
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            if(i==j) G[i][j]=0;
            else G[i][j]=INF;
        }
        valsum[i]=val[i];
    }
}


int main()
{
    scanf("%d%d%d%d",&n,&m,&st,&en);
    for(int i=0; i<n; i++) scanf("%d",&val[i]);
    inition();
    for(int i=1; i<=m; i++)
        {scanf("%d%d%d",&x,&y,&z);G[x][y]=z;G[y][x]=z;}//無向圖
    dijkstra();
    printf("%d %d\n",pathsum[en],valsum[en]);
    P();
    return 0;
}