P3371 【模板】单源最短路径(弱化版)

可紊 提交于 2019-11-29 08:19:20

P3371 【模板】单源最短路径(弱化版)

此题使用【堆优化Dijkstra求最短路模版】

AC代码

#include <iostream>
#include <cstdio>
#include <queue>
#include <utility>
#include <algorithm>
#include <vector>
#define MAXN 100010     //最大顶点数
#define inf 2147483647   //极大数据
using namespace std;
typedef pair<int, int> info;    //存放当前点及从源点到当前点的最短路径的信息
priority_queue<info, vector<info>, greater<info> > Q;
//优先队列实现将已确定最短路径的点放入队列 小顶堆实现堆优化
struct edge{
    int to;     //弧指向的点
    int weight;     //弧的权重
    edge(int t, int w): to(t), weight(w){}
};
vector<edge> G[MAXN];   //存放当前点出发的每一条弧的信息
int n, m, s;    //点 弧 源点
int dis[MAXN];  //从源点到当前点的最短路径(迪杰斯特拉跑完之前不一定是最终解)
void Dijkstra(int s){   //s为源点,使用Dijkstra查找源点到其他任意点的最短路径
    for (int i = 1; i <= n; i++) dis[i] = inf;  //初始化源点到任意其他点最短路径无穷大
    dis[s] = 0; //源点到源点最短路径为0
    Q.push(make_pair(0, s));    //将源点放入队列
    while (!Q.empty()){
        info u = Q.top();
        Q.pop();    //取出并去除队头元素
        int x = u.second;   //当前点编号
        for (int i = 0; i < G[x].size(); i++){  //遍历从x点出发的所有弧
            edge e = G[x][i];   //从x点出发的第i条弧
            if (dis[e.to] > dis[x] + e.weight){
                //若经过x点的松弛使得从源点到弧尾指向的点的最短路径更短 则更新最短路径
                dis[e.to] = dis[x] + e.weight;
                Q.push(make_pair(dis[e.to], e.to)); //将已确定最短路径的点放入队列
            }
        }
    }
}
//--------------【使用模版】-----------------
int main(){
    scanf("%d%d%d", &n, &m, &s);    //点 弧 源点
    int from, to, weight;   //弧的起点 终点 权重
    for (int i = 1; i <= m; i++){
        scanf("%d%d%d", &from, &to, &weight);
        edge in(to, weight);
        G[from].push_back(in);  //往图中加入所有已知弧的信息
    }
    Dijkstra(s);    //算法实施
    for (int i = 1; i < n; i++) printf("%d ", dis[i]);
    printf("%d\n", dis[n]);
    //输出从源点到其他所有点的最短路径
    return 0;
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!