PTA A 1030迪杰斯特拉求解两个边权的问题,

第一标尺是距离第二标尺是花费

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
#include<queue>
#include<string>
using namespace std;
#define inf 1000000000
#define maxn 1000
struct e
{
    
    
    int d;
    int cost;
};
int st,ed,n,m;
e G[maxn][maxn];
int weight[maxn];
int dis[maxn];
bool vis[maxn]={
    
    false};
vector<int>temppath,path;
vector<int>pre[maxn];
int sumc,minsum=inf;
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].d!=0){
    
    
                if(dis[u]+G[u][i].d<dis[i]){
    
    
                    dis[i]=dis[u]+G[u][i].d;
                    pre[i].clear();
                    pre[i].push_back(u);
                }
                else if(dis[u]+G[u][i].d==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--){
    
    
            sumc+=G[temppath[i]][temppath[i-1]].cost;
        }
        if(sumc<minsum){
    
    
            minsum=sumc;
            path=temppath;
        }
        sumc=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<m;i++){
    
    
        int s,e,w,c;
        scanf("%d%d%d%d",&s,&e,&w,&c);
        G[s][e].d=w;
        G[e][s].d=w;
        G[s][e].cost=c;
        G[e][s].cost=c;
    }
    dijkstra(st);
    DFS(ed);
    for(int i=path.size()-1;i>=0;i--){
    
    
        printf("%d ",path[i]);
    }
    printf("%d %d",dis[ed],minsum);
}

猜你喜欢

转载自blog.csdn.net/weixin_45890608/article/details/112576495
PTA