题目描述
如题,给出一个有向图,请输出从某一点出发到所有点的最短路径长度。
输入格式
$$ 第一行包含三个整数 n,m,s 分别表示点的个数、有向边的个数、出发点的编号。 $$
$$ 接下来 m 行每行包含三个整数 u,v,w 表示一条 u→v 的,长度为 w 的边。 $$
输出格式
$$ 输出一行 n 个整数,第 i 个表示 s 到第 i 个点的最短路径,若不能到达则输出 2^{31}−1。$$
样例
输入 #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
堆优化dijkstra
就过掉了,和AcWing 850. Dijkstra求最短路 II差不多
AcWing 850. Dijkstra求最短路 II 代码
代码:
#include<iostream>
#include<cstring>
#include<queue>
const int N = 1e4 + 10,M = 5e5 + 10;
using namespace std;
typedef pair<int,int> PII;
int n, m, s;
int h[N], e[M], ne[M], w[M], idx=0;
int dist[N];
bool st[N];
void add(int a, int b, int c)
{
e[idx] = b,w[idx] = c,ne[idx] = h[a], h[a] = idx ++;
}
void dijkstra()
{
memset(dist, 0x3f, sizeof dist);
dist[s] = 0;
priority_queue<PII, vector<PII>, greater<PII>>heap;
heap.push({0, s});
while(heap.size())
{
auto t = heap.top();
heap.pop();
int ver = t.second, distance = t.first;
if(st[ver])continue;
st[ver] = true;
for(int i = h[ver]; ~i; i = ne[i])
{
int j = e[i];
if(dist[j] > distance+w[i])
{
dist[j] = distance + w[i];
heap.push({dist[j],j});
}
}
}
}
int main()
{
scanf("%d %d %d", &n, &m, &s);
memset(h, -1, sizeof h);
while(m --)
{
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
add(a, b, c);
}
dijkstra();
for(int i = 1; i <= n; ++ i)
if(dist[i] == 0x3f3f3f3f)printf("2147483647 ");
else printf("%d ", dist[i]);
return 0;
}