結果

問題 No.2764 Warp Drive Spacecraft
ユーザー aplysiaSheepaplysiaSheep
提出日時 2024-05-19 15:17:27
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,621 bytes
コンパイル時間 1,072 ms
コンパイル使用メモリ 102,708 KB
実行使用メモリ 13,888 KB
最終ジャッジ日時 2024-05-19 15:17:37
合計ジャッジ時間 9,577 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
13,884 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 WA -
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 AC 2 ms
6,944 KB
testcase_15 AC 2 ms
6,940 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using namespace std;

struct Edge {
    int to;
    long long cost;
};

const long long INF = numeric_limits<long long>::max();

int main() {
    int N, M;
    cin >> N >> M;

    vector<long long> W(N + 1);
    for (int i = 1; i <= N; ++i) {
        cin >> W[i];
    }

    vector<vector<Edge>> graph(N + 1);

    for (int i = 0; i < M; ++i) {
        int U, V;
        long long T;
        cin >> U >> V >> T;
        graph[U].push_back({V, T});
        graph[V].push_back({U, T});
    }

    // ダイクストラ法の準備
    priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>> pq;
    vector<long long> dist(N + 1, INF);

    dist[1] = 0;
    pq.push({0, 1});

    while (!pq.empty()) {
        auto [current_dist, u] = pq.top();
        pq.pop();

        if (current_dist > dist[u]) continue;

        // 通常の航路での更新
        for (const auto& edge : graph[u]) {
            int v = edge.to;
            long long cost = edge.cost;

            if (dist[u] + cost < dist[v]) {
                dist[v] = dist[u] + cost;
                pq.push({dist[v], v});
            }
        }

        // ワープによる更新
        for (int v = 1; v <= N; ++v) {
            if (u != v) {
                long long warp_cost = W[u] * W[v];
                if (dist[u] + warp_cost < dist[v]) {
                    dist[v] = dist[u] + warp_cost;
                    pq.push({dist[v], v});
                }
            }
        }
    }

    cout << dist[N] << endl;

    return 0;
}
0