1. 程式人生 > >dijkstra+堆優化+鏈式前向星

dijkstra+堆優化+鏈式前向星

直接見題:

 

題目背景

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;
}