码迷,mamicode.com
首页 > 其他好文 > 详细

Dijkstra模板

时间:2018-10-18 19:48:35      阅读:159      评论:0      收藏:0      [点我收藏+]

标签:注意   main   范围   str   大于号   标准版   一个   end   tin   

注意几点

1.看清楚是有向还是无向图
2.dis初始值设为0x3f3f3f3f可能会不够 还是设为0x7fffffff吧,具体来说你要看题目给的范围,若是说边权和<=1e9,那么设成0x3f3f3f3f也行
3.中间有一个vis数组,其实不开这个数组,写dis[x] != d也可以,本质上是堆的懒惰删除,当一个点第一次从堆中被取出来,就说明这个点的最短路已经确定好了,此时堆中还有一些,点的编号和刚刚取出的点相同的,并且“最短路”更长,这些点显然无法更新最短路了,虽然不去有意删除这些点也可以,但是做一下洛谷的单源最短路标准版就会发现不加这个删除可能会T掉几个点。。。

 int x = q.top().n;
        int d = q.top().w;
        q.pop();
        if (d != dis[x])continue;
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
const int MAXN = 100000 + 10;
const int MAXM = 200000 + 10;
int n,m,begin,end,dis[MAXN],vis[MAXN],tot,last[MAXN],s;
struct Edge{
    int u,v,w,to;
    Edge(){}
    Edge(int u, int v, int w, int to) : u(u), v(v), w(w), to(to) {}
} e[MAXM*2];
inline void add(int u,int v,int w) {
    e[++tot] = Edge(u,v,w,last[u]);
    last[u] = tot;
}
struct st{
    int n,w;
    bool operator <(const st &a) const {
        return w > a.w;//把STL里面的小于号全部换为大于号
    }
};
std::priority_queue<st> q;
inline void dijkstra() {
    for(int i=1; i<=n; i++)
        dis[i] = 0x7fffffff;
    dis[s] = 0;
    q.push((st){s,0});
    while(!q.empty()) {
        int x = q.top().n;
        q.pop();
        if (vis[x])continue;
        vis[x] = 1;
        for(int i=last[x]; i; i=e[i].to) {
            int y = e[i].v;
            if (dis[x] + e[i].w < dis[y]){
                dis[y] = dis[x] + e[i].w;
                q.push((st){y,dis[y]});
            }
        }
    }
}
int main() {
    scanf("%d %d %d",&n,&m,&s);
    for(int i=1; i<=m; i++) {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        add(u,v,w);
    }
    dijkstra();
    for(int i=1; i<=n; i++)
        printf("%d ", dis[i]);
}

Dijkstra模板

标签:注意   main   范围   str   大于号   标准版   一个   end   tin   

原文地址:https://www.cnblogs.com/Zolrk/p/9812641.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!