A1030 Travel Plan（最短路径/Dijkstra/模板题）

题目描述

题意：N个城市，M条道路，给出M条道路的距离和花费，求从起点到终点的最短路径，若有多条最短路径，求花费最小的那条

两种做法：一种，直接Dijkstra，求出花费最小的最短路径；第二种，先Dijkstra求出所有最短路径，再用DFS求出花费最小的那条

1.Dijkstra：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=510;
const int INF=1000000000;
int n,m,st,ed;
int dis[MAXN][MAXN],cost[MAXN][MAXN];
int vis[MAXN],d[MAXN],c[MAXN],pre[MAXN];

void Dijkstra()
{
    fill(d,d+MAXN,INF);
    fill(c,c+MAXN,INF);
    d[st]=0;
    c[st]=0;
    for(int i=0;i<n;i++)
    {
        int u=-1,MIN=INF;
        for(int j=0;j<n;j++){
            if(vis[j]==-1&&d[j]<MIN)
            {
                u=j;
                MIN=d[j];
            }
        }
        if(u==-1) return;
        vis[u]=1;
        for(int v=0;v<n;v++){
            if(vis[v]==-1&&dis[u][v]!=INF){
                if(d[u]+dis[u][v]<d[v]){
                    d[v]=d[u]+dis[u][v];
                    c[v]=c[u]+cost[u][v];
                    pre[v]=u;
                }else if(d[u]+dis[u][v]==d[v]){
                    if(c[u]+cost[u][v]<c[v]){
                        c[v]=c[u]+cost[u][v];
                        pre[v]=u;
                    }
                }
            }
        }
    }
}

void DFS(int v)
{
    if(v==st){
        printf("%d ",v);
        return;
    }
    DFS(pre[v]);
    printf("%d ",v);
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&st,&ed);
    int u,v;
    fill(dis[0],dis[0]+MAXN*MAXN,INF);
    for(int i=0;i<m;i++){
        scanf("%d%d",&u,&v);
        scanf("%d%d",&dis[u][v],&cost[u][v]);
        dis[v][u]=dis[u][v];//注意此处：无向图要双向建边
        cost[v][u]=cost[u][v];
    }
    memset(vis,-1,sizeof(vis));
    Dijkstra();
    DFS(ed);
    printf("%d %d\n",d[ed],c[ed]);
    return 0;

}

2.Dijkstra+DFS：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int MAXN=510;
const int INF=1000000000;
int n,m,st,ed,mincost=INF;
int dis[MAXN][MAXN],cost[MAXN][MAXN];
int vis[MAXN],d[MAXN];
vector<int> pre[MAXN];
vector<int> tempPath,path;//临时路径及最优路径

void Dijkstra()
{
    fill(d,d+MAXN,INF);
    d[st]=0;
    for(int i=0;i<n;i++)
    {
        int u=-1,MIN=INF;
        for(int j=0;j<n;j++){
            if(vis[j]==-1&&d[j]<MIN)
            {
                u=j;
                MIN=d[j];
            }
        }
        if(u==-1) return;
        vis[u]=1;
        for(int v=0;v<n;v++){
            if(vis[v]==-1&&dis[u][v]!=INF){
                if(d[u]+dis[u][v]<d[v]){
                    d[v]=d[u]+dis[u][v];
                    pre[v].clear();
                    pre[v].push_back(u);
                }else if(d[u]+dis[u][v]==d[v]){
                    pre[v].push_back(u);
                }
            }
        }
    }
}

void DFS(int v)
{
    if(v==st){
        tempPath.push_back(v);
        int tempcost=0;
        for(int i=tempPath.size()-1;i>0;i--){
            int id=tempPath[i],idNext=tempPath[i-1];
            tempcost+=cost[id][idNext];
        }
        if(tempcost<mincost){
            mincost=tempcost;
            path=tempPath;
        }
        tempPath.pop_back();
        return;
    }
    tempPath.push_back(v);
    for(int i=0;i<pre[v].size();i++)
        DFS(pre[v][i]);
    tempPath.pop_back();
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&st,&ed);
    int u,v;
    fill(dis[0],dis[0]+MAXN*MAXN,INF);
    for(int i=0;i<m;i++){
        scanf("%d%d",&u,&v);
        scanf("%d%d",&dis[u][v],&cost[u][v]);
        dis[v][u]=dis[u][v];
        cost[v][u]=cost[u][v];
    }
    memset(vis,-1,sizeof(vis));
    Dijkstra();
    DFS(ed);
    for(int i=path.size()-1;i>=0;i--)
        printf("%d ",path[i]);
    printf("%d %d\n",d[ed],mincost);
    return 0;

}