堆优化Dijkstra(Luogu 4779)

• Pair

#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;

int n,m,s,dis[100005],head[100005],cnt=0;
bool vis[100005];


struct edge{
    int to,w,next;
}e[500005];


inline void add(int u,int v,int w){
    e[++cnt].w=w;
    e[cnt].to=v;
    e[cnt].next=head[u]; 
    head[u]=cnt;
}

priority_queue<pair<int,int> > q;
inline void dijkstra(){
    for(int i=1; i<=n; i++) dis[i]=0x7fffffff;
    dis[s]=0;
    q.push(make_pair(0,s));
    while(!q.empty()){
        int x=q.top().second;
        q.pop();
        if(vis[x]) continue;
        vis[x]=1;
        for(int i=head[x];i!=-1;i=e[i].next){
            int y=e[i].to;
            if(!vis[y]&&dis[y]>dis[x]+e[i].w){
                dis[y]=dis[x]+e[i].w;
                q.push(make_pair(-dis[y],y));
            }
        }
    }
}

int main(){
    memset(head,-1,sizeof(head));
    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]);
}

• Struct

#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;

int n,m,s,dis[100005],head[100005],cnt=0;
bool vis[100005];


struct edge{
    int to,w,next;
}e[500005];


inline void add(int u,int v,int w){
    e[++cnt].w=w;
    e[cnt].to=v;
    e[cnt].next=head[u]; 
    head[u]=cnt;
}
struct node{
    int dis,u;
    bool operator <(const node &x)const{//
        return x.dis<dis;
    }
};
priority_queue<node> q;
inline void dijkstra(){
    for(int i=1; i<=n; i++) dis[i]=0x7fffffff;
    dis[s]=0;
    q.push((node){0,s});
    while(!q.empty()){
        node t=q.top();
        q.pop();
        int x=t.u;
        if(vis[x]) continue;
        vis[x]=1;
        for(int i=head[x];i!=-1;i=e[i].next){
            int y=e[i].to;
            if(!vis[y]&&dis[y]>dis[x]+e[i].w){
                dis[y]=dis[x]+e[i].w;
                q.push((node){dis[y],y});
            }
        }
    }
}

int main(){
    memset(head,-1,sizeof(head));
    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]);
}