Dijkstra求最短路的条数,并输出最短路径和最短路经过的点的最大和

时间:2022-11-25 22:19:42
#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; }