题目背景
本题测试数据为随机数据,在考试中可能会出现构造数据让SPFA不通过,如有需要请移步 P4779。
题目描述
如题,给出一个有向图,请输出从某一点出发到所有点的最短路径长度。
输入输出格式
输入格式:
第一行包含三个整数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。保证数据随机。
对于真正 100% 的数据,请移步 P4779。请注意,该题与本题数据范围略有不同。
样例说明:
图片1到3和1到4的文字位置调换
AC代码 Dijkstra + 堆优化
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 10001;
const int maxm = 500001;
const int inf = 2147483647;
typedef pair<int,int> pa;
struct edge{
int to,next,v;
}e[maxm];
int head[maxn],dis[maxn],n,m,num_edge,s;
bool vis[maxn];
void Dijkstra(int s){
priority_queue<pa,vector<pa>,greater<pa> > q;
q.push(make_pair(0,s));
for(int i = 1;i <= n;i++)
dis[i] = inf;
dis[s] = 0;
while(!q.empty()){
int u = q.top().second;
q.pop();
if(vis[u]) continue;
vis[u] = 1;
for(int i = head[u];i;i = e[i].next){
if(dis[u] + e[i].v < dis[e[i].to] && !vis[e[i].to]){
dis[e[i].to] = dis[u] + e[i].v;
q.push(make_pair(dis[e[i].to],e[i].to));
}
}
}
}
void add(int from,int to,int v){
e[++num_edge].next = head[from];
e[num_edge].to = to;
e[num_edge].v = v;
head[from] = num_edge;
}
int main(){
ios::sync_with_stdio(false);
cin>>n>>m>>s;
for(int i = 1;i <= m;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
}
Dijkstra(s);
for(int i = 1;i <= n;i++) cout<<dis[i]<<" ";
return 0;
}