結果

問題 No.1449 新プロランド
ユーザー rogi52
提出日時 2022-10-07 21:27:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,566 bytes
コンパイル時間 2,185 ms
コンパイル使用メモリ 207,504 KB
最終ジャッジ日時 2025-02-07 22:50:06
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); i++)
using namespace std;
typedef long long ll;

// g <- pair < v , cost > 
template < class T >
vector< T > dijkstra(vector<vector<pair<int, T>>> &graph, int s) {
    T INF = numeric_limits< T >::max();
    vector<T> dist(graph.size(), INF);
    priority_queue<pair<T,int>, vector<pair<T,int>>, greater<pair<T,int>>> q;
    q.push({dist[s] = T(0), s});
    while(!q.empty()){
        auto [uc, ui] = q.top(); q.pop();
        if(uc != dist[ui]) continue;
        for(auto [vi, vc] : graph[ui]) if(dist[vi] > uc + vc) 
            q.push({dist[vi] = uc + vc, vi});
    }
    return dist;
}

int main(){
    cin.tie(0);
    ios::sync_with_stdio(0);
    
    int N,M; cin >> N >> M;
    vector<tuple<int,int,int>> E(M);
    for(auto &[a, b, c] : E) cin >> a >> b >> c, a--, b--;
    vector<int> T(N);
    rep(i,N) cin >> T[i];

    int MAX_C = 1000;
    vector<vector<pair<int,int>>> G(N * (MAX_C + 1));
    auto h = [MAX_C](int v, int p) { return v * (MAX_C + 1) + p; };

    for(auto &[a, b, c] : E) {
        for(int p = 0; p <= MAX_C; p++) {
            if(p + T[a] != 0)
                G[h(a, p)].push_back({h(b, min(MAX_C, p + T[a])), c / min(MAX_C, p + T[a]) + T[a]});
            if(p + T[b] != 0)
                G[h(b, p)].push_back({h(a, min(MAX_C, p + T[b])), c / min(MAX_C, p + T[b]) + T[b]});
        }
    }

    auto dist = dijkstra(G, h(0, 0));
    int ans = numeric_limits<int>::max();
    for(int p = 0; p <= MAX_C; p++) ans = min(ans, dist[h(N - 1, p)]);
    cout << ans << endl;
}
0