用迪杰斯特拉算法求解单源最短路径,有多条最短路径时,求第二标杆最大的路径(可以用DFS实现)
首先迪杰斯特拉算法描述如下
先初始化
初始时除起点到起点的距离为0外,起点到其他点的距离为inf一个很大的数,
然后在起点到其他点的距离中寻找距离最短的,将这个点标记为以访问,然后通过这个中介点到其余未访问的点的距离如果比原来记录的短,则更新这条记录,重复顶点数次循环,最终可以得到起点到各个顶点的最短距离
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
#include<queue>
#include<string>
using namespace std;
#define inf 1000000000
#define maxn 1000
int st,ed,n,m;
int G[maxn][maxn];
int weight[maxn];
int dis[maxn];
bool vis[maxn]={
false};
vector<int>temppath;
vector<int>pre[maxn];
int cnt,sumw,maxsum;
void dijkstra(int st)
{
fill(dis,dis+maxn,inf);
dis[st]=0;
for(int i=0;i<n;i++){
int u=-1,dmin=inf;
for(int v=0;v<n;v++){
if(dis[v]<dmin&&vis[v]==false){
u=v;
dmin=dis[v];
}
}
if(u==-1){
return ;
}
vis[u]=true;
for(int i=0;i<n;i++){
if(vis[i]==false&&G[u][i]!=0){
if(dis[u]+G[u][i]<dis[i]){
dis[i]=dis[u]+G[u][i];
pre[i].clear();
pre[i].push_back(u);
}
else if(dis[u]+G[u][i]==dis[i]){
pre[i].push_back(u);
}
}
}
}
}
void DFS(int root)
{
if(root==st){
temppath.push_back(root);
for(int i=temppath.size()-1;i>=0;i--){
sumw+=weight[temppath[i]];
}
if(sumw>maxsum){
maxsum=sumw;
}
cnt++;
sumw=0;
temppath.pop_back();
return;
}
temppath.push_back(root);
for(int i=0;i<pre[root].size();i++){
DFS(pre[root][i]);
}
temppath.pop_back();
}
int main()
{
scanf("%d %d %d %d",&n,&m,&st,&ed);
for(int i=0;i<n;i++){
scanf("%d",&weight[i]);
}
for(int i=0;i<m;i++){
int s,e,w;
scanf("%d%d%d",&s,&e,&w);
G[s][e]=w;
G[e][s]=w;
}
dijkstra(st);
DFS(ed);
printf("%d %d",cnt,maxsum);
}