PAT 甲级 1030 Travel Plan

PAT 甲级 1030 Travel Plan

原题链接

算法标签 图论 单源最短路dijkstra

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 505;
// 存储路径 存储花费 
int d[N][N],c[N][N];
// 存储最短路径 存储最低花费 存储前驱路径 
int dist[N], cost[N], pre[N];
bool st[N];
int n,m,s,t;
void dij(){
    memset(dist, 0x3f, sizeof dist);
    memset(cost, 0x3f, sizeof cost);
    dist[s]=0, cost[s]=0;
    for(int i=0;i<n;++i){
        int t=-1;
        for(int j=0;j<n;++j){
            if(!st[j]&&(t==-1||dist[t]>dist[j])){
                t=j;
            }
        }
        st[t]=true;
        for(int j=0;j<n;++j){
            if(dist[j]>dist[t]+d[t][j]){
                dist[j]=dist[t]+d[t][j];
                cost[j]=cost[t]+c[t][j];
                pre[j]=t;
            }
            else if(dist[j]==dist[t]+d[t][j]&&cost[j]>cost[t]+c[t][j]){
                cost[j]=cost[t]+c[t][j];
                pre[j]=t;
            }
        }
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m>>s>>t;
    memset(d, 0x3f, sizeof d);
    memset(c, 0x3f, sizeof c);

    while(m--){
        int a,b,cc,dd;
        cin>>a>>b>>cc>>dd;
        // 若c比d[a][b]大，但是d比c[a][b]小的话，d[a][b]没有更新，但是c[a][b]被更新了，这样d,c里面存的不就不是同一条路径的信息
        //d[a][b]=d[b][a]=min(d[a][b], c);
        //c[a][b]=c[b][a]=min(c[a][b], d);
        if(cc<d[a][b]){
            d[a][b]=d[b][a]=cc;
            c[a][b]=c[b][a]=dd;
        }
        else if(cc==d[a][b]&&dd<c[a][b]){
            c[a][b]=c[b][a]=dd;
        }
    }
    dij();
    vector<int> path;
    for(int i=t;i!=s;i=pre[i]){
        path.push_back(i);    
    }
    cout<<s;
    for(int i=path.size()-1;i>=0;--i){
        cout<<" "<<path[i];    
    }
    cout<<" "<<dist[t]<<" "<<cost[t];
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈