PTA A 1003 迪杰斯特拉算法加DFS实现最短路径和最大点权和

用迪杰斯特拉算法求解单源最短路径,有多条最短路径时,求第二标杆最大的路径(可以用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);
}

猜你喜欢

转载自blog.csdn.net/weixin_45890608/article/details/112572788