結果
問題 |
No.1308 ジャンプビーコン
|
ユーザー |
![]() |
提出日時 | 2020-12-04 20:44:02 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 519 ms / 4,000 ms |
コード長 | 2,166 bytes |
コンパイル時間 | 2,443 ms |
コンパイル使用メモリ | 206,028 KB |
最終ジャッジ日時 | 2025-01-16 16:05:19 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
#include<bits/stdc++.h> using namespace std; using P = pair<int, int>; const long long LM = 1LL << 60; long long get_dist(int s, int g, const vector<vector<P>>& edge) { if (s == g) return 0; int n = edge.size(); vector<long long> dist(n, LM); dist[s] = 0; queue<int> q; q.push(s); while (true) { int p = q.front(); if (p == g) return dist[g]; q.pop(); for (auto& e : edge[p]) { if (dist[e.second] == LM) { dist[e.second] = dist[p] + e.first; q.push(e.second); } } } } void calc(int s, long long l, long long ds, vector<long long>& dp, const vector<vector<P>>& edge) { int n = edge.size(); vector<long long> dist(n, LM); vector<bool> vis(n, false); stack<int> st; st.push(s); dist[s] = l; while (!st.empty()) { int p = st.top(); if (vis[p]) { st.pop(); for (auto& e : edge[p]) { if (dist[p] < dist[e.second]) { dp[p] = min(dp[p], dp[e.second]); } } } else { vis[p] = true; dp[p] += min(dist[p], ds); for (auto& e : edge[p]) { if (dist[e.second] == LM) { dist[e.second] = dist[p] + e.first; st.push(e.second); } } } } } int main() { cin.tie(0); ios::sync_with_stdio(0); int n, q, c; cin >> n >> q >> c; vector<vector<P>> edge(n); for (int i = 0; i < n - 1; ++i) { int u, v, l; cin >> u >> v >> l; --u; --v; edge[u].emplace_back(l, v); edge[v].emplace_back(l, u); } vector<long long> dp(n, LM); int prev; cin >> prev; --prev; dp[prev] = 0; for (int i = 1; i < q; ++i) { int x; cin >> x; --x; long long d = get_dist(prev, x, edge); calc(x, c, d, dp, edge); prev = x; } long long ans = LM; for (int i = 0; i < n; ++i) { ans = min(ans, dp[i]); } cout << ans << '\n'; return 0; }