結果

問題 No.807 umg tours
ユーザー neko_the_shadow
提出日時 2019-05-17 11:04:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,374 bytes
コンパイル時間 1,455 ms
コンパイル使用メモリ 93,596 KB
最終ジャッジ日時 2025-01-07 03:43:00
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 11 MLE * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<vector>
#include<queue>
#include<tuple>
#include<limits>

int main() {
    int n, m;
    std::cin >> n >> m;
    std::vector<std::vector<long>> costs(n, std::vector<long>(n, -1));
    for (int i = 0; i < m; i++) {
        int a, b;
        long c;
        std::cin >> a >> b >> c;

        a--;
        b--;
        costs[a][b] = c;
        costs[b][a] = c;
    }

    std::vector<long> d1(n, std::numeric_limits<long>::max());
    std::priority_queue<std::tuple<long, int>> q1;
    d1[0] = 0;
    q1.push(std::make_tuple(0, 0));
    while (!q1.empty()) {
        auto t = q1.top();
        q1.pop();

        int u = std::get<1>(t);
        for (int v = 0; v < n; v++) {
            if (costs[u][v] == -1) {
                continue;
            }

            if (d1[u] + costs[u][v] < d1[v]) {
                d1[v] = d1[u] + costs[u][v];
                q1.push(std::make_tuple(d1[v], v));
            }
        }
    }

    std::vector<long> d2(n, std::numeric_limits<long>::max()); // NOT SKIPPED
    std::vector<long> d3(n, std::numeric_limits<long>::max()); // SKIPPED
    std::priority_queue<std::tuple<long, int, bool>> q2;
    d2[0] = 0;
    d3[0] = 0;
    q2.push(std::make_tuple(0, 0, false));
    q2.push(std::make_tuple(0, 0, true));

    while (!q2.empty()) {
        auto t = q2.top();
        q2.pop();

        int u = std::get<1>(t);
        bool skipped = std::get<2>(t);
        if (skipped) {
            for (int v = 0; v < n; v++) {
                if (costs[u][v] == -1) {
                    continue;
                }

                if (d3[u] + costs[u][v] < d3[v]) {
                    d3[v] = d3[u] + costs[u][v];
                    q2.push(std::make_tuple(d3[v], v, true));
                }
            }
        } else {
            for (int v = 0; v < n; v++) {
                if (costs[u][v] == -1) {
                    continue;
                }

                if (d2[u] + costs[u][v] < d2[v]) {
                    d2[v] = d2[u] + costs[u][v];
                    q2.push(std::make_tuple(d2[v], v, false));
                }

                if (d2[u] < d3[v]) {
                    d3[v] = d2[u];
                    q2.push(std::make_tuple(d3[v], v, true));
                }
            }
        }
    }

    for (int i = 0; i < n; i++) {
        std::cout << d1[i] + d3[i] << std::endl;
    }
}
0