結果

問題 No.788 トラックの移動
コンテスト
ユーザー rpy3cpp
提出日時 2019-05-05 00:36:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,820 bytes
コンパイル時間 2,253 ms
コンパイル使用メモリ 183,168 KB
実行使用メモリ 34,688 KB
最終ジャッジ日時 2024-06-23 13:32:15
合計ジャッジ時間 5,811 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
constexpr long long INF = 1e18;

struct Edge{
    int to;
    long long cost;
    Edge(int to, long long cost):to(to), cost(cost) {}
};

void dijkstra(int start, vector<long long> &dist, const vector<vector<Edge>> &Es){
    int N = Es.size();
    dist.assign(N, INF);
    dist[start] = 0;
    priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>> pq;
    pq.push(make_pair(0LL, start));
    while (not pq.empty()){
        auto dv = pq.top();
        pq.pop();
        if (dv.first > dist[dv.second]) continue;
        dist[dv.second] = dv.first;
        for (auto &uc : Es[dv.second]){
            long long new_d = dv.first + uc.cost;
            if (new_d < dist[uc.to]){
                dist[uc.to] = new_d;
                pq.push(make_pair(new_d, uc.to));
            }
        }
    }
}


int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, M, L;
    cin >> N >> M >> L;
    vector<int> Ts(N, 0);
    for (auto & t : Ts) cin >> t;
    vector<vector<Edge>> Es(N, vector<Edge>());
    for (int i = 0; i < M; ++i){
        int a, b;
        long long c;
        cin >> a >> b >> c;
        Es[a - 1].push_back({b - 1, c});
        Es[b - 1].push_back({a - 1, c});
    }
    vector<vector<long long>> dist(N, vector<long long>());
    for (int v = 0; v < N; ++v) dijkstra(v, dist[v], Es);
    long long ans = INF;
    for (int g = 0; g < N; ++g){
        long long len = 0LL;
        for (int i = 0; i < N; ++i) len += dist[g][i] * 2 * Ts[i];
        long long dif = INF;
        for (int j = 0; j < N; ++j) {
            if (Ts[j] >= 1){
                dif = min(dif, dist[L - 1][j] - dist[j][g]);
            }
        }
        ans = min(ans, len + dif);
    }
    cout << ans << endl;
    return 0;
}
0