洛谷 P3371 【模板】单源最短路径(Dijkstra + 堆优化)

来源:互联网 发布:数组长度是什么 编辑:程序博客网 时间:2024/06/04 18:19

题目描述
如题,给出一个有向图,请输出从某一点出发到所有点的最短路径长度。
输入输出格式
输入格式:
第一行包含三个整数N、M、S,分别表示点的个数、有向边的个数、出发点的编号。
接下来M行每行包含三个整数Fi、Gi、Wi,分别表示第i条有向边的出发点、目标点和长度。
输出格式:
一行,包含N个用空格分隔的整数,其中第i个整数表示从点S出发到点i的最短路径长度(若S=i则最短路径长度为0,若从点S无法到达点i,则最短路径长度为2147483647)
输入输出样例
输入样例#1: 复制
4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4
输出样例#1: 复制
0 2 4 3
说明
时空限制:1000ms,128M
数据规模:
对于20%的数据:N<=5,M<=15
对于40%的数据:N<=100,M<=10000
对于70%的数据:N<=1000,M<=100000
对于100%的数据:N<=10000,M<=500000
样例说明:
这里写图片描述

#include<iostream>#include<cstring>#include<algorithm>#include<queue>#include<cstdio>#include<vector>#define MAXN 10005#define MAXM 500005using namespace std;struct Edge{ int to,nxt,w; }e[MAXM << 1];struct Node{    int u,dis;    inline bool operator > (const Node &b) const {        return dis > b.dis;    }};priority_queue<Node,vector<Node>,greater<Node> >q;int tot,head[MAXN],dis[MAXN];bool vis[MAXN];inline void Add_Edge(int u,int v,int w) {    e[++tot].to = v,e[tot].w = w,e[tot].nxt = head[u],head[u] = tot;    //e[++tot].to = u,e[tot].w = w,e[tot].nxt = head[v],head[v] = tot;}inline void read(int &x) {    x = 0; register char c = getchar();    while(!isdigit(c)) c = getchar();    while(isdigit(c)) x = x * 10 + c - '0',c = getchar();}int main(int argc,char *argv[]) {    int n,m,S,u,v,w; read(n),read(m),read(S);    for(int i=1; i<=m; ++i) {        read(u),read(v),read(w);        Add_Edge(u,v,w);    }    memset(dis,127 / 3,sizeof dis );    int INF = dis[0];    memset(vis,0,sizeof vis );    Node Now;    q.push((Node){S,0});    dis[S] = 0;    while(!q.empty()) {        Now = q.top(); q.pop();        if(vis[Now.u]) continue;        int u = Now.u;  vis[u] = 1;        for(int i=head[u]; i; i=e[i].nxt) {            int v = e[i].to;            if(dis[v] > dis[u] + e[i].w) {                dis[v] = dis[u] + e[i].w;                q.push((Node){v,dis[v]});            }        }    }    for(int i=1; i<=n; ++i){        if(dis[i] == INF) dis[i] = 2147483647;        printf("%d ",dis[i]);    }    return 0;}
原创粉丝点击