旅游规划 (25 分)(dijkstra+Floyd两种算法三种写法)

有了一张自驾旅游路线图，你会知道城市间的高速公路长度、以及该公路要收取的过路费。现在需要你写一个程序，帮助前来咨询的游客找一条出发地和目的地之间的最短路径。如果有若干条路径都是最短的，那么需要输出最便宜的一条路径。

输入格式:

输入说明：输入数据的第1行给出4个正整数N、M、S、D，其中N（2≤N≤500）是城市的个数，顺便假设城市的编号为0~(N−1)；M是高速公路的条数；S是出发地的城市编号；D是目的地的城市编号。随后的M行中，每行给出一条高速公路的信息，分别是：城市1、城市2、高速公路长度、收费额，中间用空格分开，数字均为整数且不超过500。输入保证解的存在。

输出格式:

在一行里输出路径的长度和收费总额，数字间以空格分隔，输出结尾不能有多余空格。

输入样例:

4 5 0 3
0 1 1 20
1 3 2 30
0 3 4 10
0 2 2 20
2 3 1 20

输出样例:

3 40

Floyd：时间复杂度O（N^3）

#include<bits/stdc++.h>

using namespace std;

const int N=501,inf=0x3f3f3f3f;

int g[N][N],w[N][N];
int n,m,s,d;

void floyd()
{
    for(int k=0;k<n;k++)
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                if(g[i][j]>g[i][k]+g[k][j]||(g[i][j]==g[i][k]+g[k][j]&&w[i][j]>w[i][k]+w[k][j]))
                {
                    g[i][j]=g[i][k]+g[k][j];
                    w[i][j]=w[i][k]+w[k][j];
                }
}
int main()
{
    cin>>n>>m>>s>>d;
    memset(g,inf,sizeof g);
    memset(w,inf,sizeof w);
    while( m-- )
    {
        int a,b,l,m;
        cin>>a>>b>>l>>m;
        g[a][b]=g[b][a]=l;
        w[a][b]=w[b][a]=m;
    }
    floyd();
    cout<<g[s][d]<<' '<<w[s][d];
    return 0;
}

dijkstra() :O(N^2)

#include<bits/stdc++.h>

using namespace std;

const int N=510,inf=0x3f3f3f3f;

int dist[N],vis[N],money[N];
int g[N][N],w[N][N];
int n,m,s,d;

void dijkstra()
{
    memset(dist,inf,sizeof dist);
    memset(money,inf,sizeof money);
    dist[s]=0;
    money[s]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=0;j<n;j++)
            if(!vis[j]&&(t==-1||dist[j]<dist[t]))
                t=j;
        vis[t]=1;
        for(int j=0;j<n;j++)
        {
            if(dist[j]>dist[t]+g[t][j])
            {
                dist[j]=dist[t]+g[t][j];
                money[j]=money[t]+w[t][j];
            }
            else if(dist[j]==dist[t]+g[t][j]&&money[j]>money[t]+w[t][j])
                money[j]=money[t]+w[t][j];
        }
    }
}
int main()
{
    cin>>n>>m>>s>>d;
    memset(g,inf,sizeof g);
    memset(w,inf,sizeof w);
    while( m-- )
    {
        int a,b,l,m;
        cin>>a>>b>>l>>m;
        g[a][b]=g[b][a]=l;
        w[a][b]=w[b][a]=m;
    }
    dijkstra();
    cout<<dist[d]<<' '<<money[d];
    return 0;
}

堆优化版dijkstra: O(MlogN)

 

#include<bits/stdc++.h>

#define x first
#define y second

using namespace std;

typedef pair<pair<int,int>,int>PII;

const int N=1000010,inf=0X3f3f3f3f;

int h[N],e[N],w[N],m[N],ne[N],idx;
int n,mm,s,d,dist[N],money[N],vis[N];

void add(int a,int b,int c,int d)//邻接表
{
    e[idx]=b,w[idx]=c,m[idx]=d,ne[idx]=h[a],h[a]=idx++;
}
void dijkstra()
{
    memset(dist,inf,sizeof dist);
    memset(money,inf,sizeof money);
    dist[s]=0,money[s]=0;
    priority_queue<PII,vector<PII>,greater<PII> >heap;
    heap.push({{0,0},s});
    while(heap.size())
    {
        auto t=heap.top();
        heap.pop();
        if(vis[t.y]) continue;
        vis[t.y]=1;
        for(int i=h[t.y];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>t.x.x+w[i]||(dist[j]==t.x.x+w[i]&&money[j]>t.x.y+m[i]))
            {
                dist[j]=t.x.x+w[i];
                money[j]=t.x.y+m[i];
                heap.push({{dist[j],money[j]},j});
            }
        }
    }
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>mm>>s>>d;
    while(mm -- )
    {
       int a,b,c,d;
       cin>>a>>b>>c>>d;
       add(a,b,c,d);
       add(b,a,c,d);
    }
    dijkstra();
    cout<<dist[d]<<' '<<money[d];
    return 0;
}