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

dijkstra (模板)

时间:2018-10-02 20:10:11      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:head   pen   node   main   ==   pre   display   str   end   

突然意识到以前写的都是假的dij,感谢GhostCai神犇。

技术分享图片
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define int long long

using namespace std;
const int MAXN = 10005;
const int MAXM = 500005;
const int inf = 2147483647;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch==-?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-0;ch=getchar();}
    return f?x:-x;
}

int n,m,S,dis[MAXN],head[MAXN],cnt;
int to[MAXM],nxt[MAXM],val[MAXM];
bool vis[MAXN];

struct Node{
    int id,w;
    friend bool operator<(const Node A,const Node B){
        return A.w>B.w;
    }
};

inline void add(int bg,int ed,int w){
    to[++cnt]=ed,nxt[cnt]=head[bg],head[bg]=cnt,val[cnt]=w;
}

priority_queue<Node> Q;

void dijkstra(){
    for(int i=1;i<=n;i++) dis[i]=inf;
    Node now;now.id=S;now.w=0;dis[S]=0;
    int p,ww;Q.push(now);
    while(!Q.empty()){
        Node x=Q.top();Q.pop();p=x.id;ww=x.w;
        if(vis[p] || dis[p]!=ww) continue;
        vis[p]=1;int u;
        for(register int i=head[p];i;i=nxt[i]){
            u=to[i];
            if(dis[p]+val[i]<dis[u]) {
                dis[u]=dis[p]+val[i];
                now.id=u;now.w=dis[u];
                Q.push(now);
            }
        }
    }    
}

signed main(){
    n=rd(),m=rd(),S=rd();int x,y,z;
    for(int i=1;i<=m;i++){
        x=rd(),y=rd(),z=rd();
        add(x,y,z);
    }
    dijkstra();
    for(int i=1;i<=n;i++) printf("%lld ",dis[i]);
    return 0;
}
View Code

 

dijkstra (模板)

标签:head   pen   node   main   ==   pre   display   str   end   

原文地址:https://www.cnblogs.com/sdfzsyq/p/9737531.html

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