Dijkstra模板

Posted 2021-01-13 zolrk

注意几点

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]);
}