pta7-7旅游规划(dijkstra算法)

题目链接:https://pintia.cn/problem-sets/1101307589335527424/problems/1101314114762387456

题意:给n给城市,m条公路,公路是双向的,起点S,终点D,并给出每条公路连接的两个city的编号以及路费,求S到D的最短距离,若具有最短距离的路线不止一条,求出路费最少的一条。

思路:单源最短路径问题,利用dijkstra算法求出S到其他点的最短路径,只不过加一个路费cost,在dis相等的情况下,取cost最小值。

AC代码:

#include<bits/stdc++.h>
using namespace std;

const int inf=0x3f3f3f3f;
int n,m,S,D;
int dis[505],cost[505],a[505][505],b[505][505],vis[505];

void dijkstra(){
    vis[S]=1;
    for(int i=0;i<n-1;++i){
        int Min=inf,k;
        for(int i=0;i<n;++i)
            if(!vis[i]&&dis[i]<Min)
                Min=dis[i],k=i;
        if(Min==inf) break;
        vis[k]=1;
        for(int i=0;i<n;++i)
            if(!vis[i]&&dis[i]>dis[k]+a[k][i]){
                dis[i]=dis[k]+a[k][i];
                cost[i]=cost[k]+b[k][i];
            }
            else if(!vis[i]&&dis[i]==dis[k]+a[k][i]&&cost[i]>cost[k]+b[k][i])
                cost[i]=cost[k]+b[k][i];
    }
}

int main(){
    scanf("%d%d%d%d",&n,&m,&S,&D);
    for(int i=0;i<n;++i)
        for(int j=0;j<n;++j)
            a[i][j]=b[i][j]=inf;
    int t1,t2,t3,t4;
    while(m--){
        scanf("%d%d%d%d",&t1,&t2,&t3,&t4);
        a[t1][t2]=a[t2][t1]=t3;
        b[t1][t2]=b[t2][t1]=t4;
    }
    for(int i=0;i<n;++i)
        dis[i]=a[S][i],cost[i]=b[S][i];
    dis[S]=cost[S]=0;
    dijkstra();
    printf("%d %d\n",dis[D],cost[D]);
    return 0;
}

 

转载于:https://www.cnblogs.com/FrankChen831X/p/10492723.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Dijkstra算法是一种用于解决带权图的单源最短路径问题的贪心算法。它维护一个距离起点的最短路径已知的顶点集合,通过不断地扩展这个集合,最终得到从起点到所有顶点的最短路径。 Dijkstra算法的基本思想是,维护一个集合S,表示已经求出最短路径的顶点集合。一开始,S只包含起点。然后,每次从集合V-S中选取一个距离起点最近的顶点u,将其加入集合S中,并更新与u相邻的所有顶点的最短路径。 具体实现上,我们可以使用一个数组dis[]来存储每个顶点到起点的最短路径长度,数组vis[]表示该顶点是否已经被加入到集合S中。每次选取距离起点最近的顶点u后,我们遍历u的所有邻居v,并更新dis[v]的值,如果dis[v]发生了改变,我们就将v加入到一个优先队列中,等待下一次选择。 以下是Dijkstra算法的伪代码实现: ``` int n; // 顶点数 int dis[N]; // 存储起点到每个顶点的最短距离 bool vis[N]; // 标记每个顶点是否已经加入集合S中 vector<pair<int, int>> adj[N]; // 存储每个顶点的邻居 void dijkstra(int s) { // s为起点编号 memset(dis, 0x3f, sizeof dis); // 将dis数组初始化为无穷大 dis[s] = 0; // 起点到自身的距离为0 priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q; q.push({0, s}); // 将起点加入队列中 while (!q.empty()) { auto t = q.top(); q.pop(); int u = t.second; if (vis[u]) continue; // 如果该点已经在集合S中,直接跳过 vis[u] = true; // 将u加入集合S中 for (auto [v, w] : adj[u]) { // 遍历u的所有邻居 if (dis[v] > dis[u] + w) { // 如果从u到v的距离更短 dis[v] = dis[u] + w; // 更新dis数组 q.push({dis[v], v}); // 将v加入队列中 } } } } ``` 其中,priority_queue是一个优先队列,用于存储待选顶点。我们使用了STL中的pair来表示顶点与其到起点的距离。优先队列默认按照pair的第一个元素排序,因此我们需要自定义一个比较函数,将pair按照第二个元素(距离)排序。 Dijkstra算法的时间复杂度为O(ElogV),其中E为边数,V为顶点数。在实际应用中,Dijkstra算法的效率很高,能够处理大规模的图。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值