dijkstra+堆優化+鏈式前向星
阿新 • • 發佈:2018-12-13
直接見題:
題目背景
2018 年 7 月 19 日,某位同學在 NOI Day 1 T1 歸程 一題裡非常熟練地使用了一個廣為人知的演算法求最短路。
然後呢?
100 \rightarrow 60100→60;
Ag \rightarrow CuAg→Cu;
最終,他因此沒能與理想的大學達成契約。
小 F 衷心祝願大家不再重蹈覆轍。
題目描述
給定一個 NN 個點,MM 條有向邊的帶非負權圖,請你計算從 SS 出發,到每個點的距離。
資料保證你能從 SS 出發到任意點。
輸入輸出格式
輸入格式:
第一行為三個正整數 N, M, SN,M,S。 第二行起 MM 行,每行三個非負整數 u_i, v_i, w_iui,vi,wi,表示從 u_iui 到 v_ivi 有一條權值為 w_iwi 的邊。
輸出格式:
輸出一行 NN 個空格分隔的非負整數,表示 SS 到每個點的距離。
輸入輸出樣例
輸入樣例#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
說明
樣例解釋請參考 資料隨機的模板題。
1 \leq N \leq 1000001≤N≤100000;
1 \leq M \leq 2000001≤M≤200000;
S = 1S=1;
1 \leq u_i, v_i\leq N1≤ui,vi≤N;
0 \leq w_i \leq 10 ^ 90≤wi≤109,
0 \leq \sum w_i \leq 10 ^ 90≤∑wi≤109。
#include<iostream> #include<queue> using namespace std; const int maxn = 4000010; const long long inf = 2147483647; struct node{ int to; int next; int w; }edge[maxn]; struct Node{ int dis; int pos; bool operator < (const Node &x) const { return dis > x.dis; } }; int head[maxn], d[maxn]; int vis[maxn]; int n, m, a, b, cnt = 1, s, w; void insert(int u, int v, int w) { edge[cnt].to = v; edge[cnt].next = head[u]; edge[cnt].w = w; head[u] = cnt++; } void dijkstra(int s){ fill(d, d + maxn, inf); d[s] = 0; priority_queue<Node> q; q.push((Node){0, s}); while (!q.empty()) { Node temp = q.top(); q.pop(); int dis = temp.dis, u = temp.pos; if (vis[u]) continue; vis[u] = 1; for (int i = head[u]; i != 0; i = edge[i].next) { int distance = edge[i].w; int v = edge[i].to; if (d[u] + distance < d[v]) { d[v] = d[u] + distance; if (!vis[v]) { q.push((Node){d[v], v}); } } } } } int main() { cin >> n >> m >> s; for (int i = 0; i < m; i++) { cin >> a >> b >> w; insert(a, b, w); } dijkstra(s); for (int i = 1; i <= n; i++) { if (i != n) cout << d[i] << " "; else cout << d[i]; } return 0; }