結果
問題 |
No.1308 ジャンプビーコン
|
ユーザー |
![]() |
提出日時 | 2020-12-02 01:29:41 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 694 ms / 4,000 ms |
コード長 | 1,878 bytes |
コンパイル時間 | 2,435 ms |
コンパイル使用メモリ | 208,536 KB |
最終ジャッジ日時 | 2025-01-16 12:46:36 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 37 |
ソースコード
#include<bits/stdc++.h> using namespace std; using P = pair<long long, int>; const long long LM = 1LL << 60; vector<vector<P>> edge; vector<long long> d; long long dfs(int p, int par) { long long mi = d[p]; for (auto& e : edge[p]) { if (e.second == par) continue; mi = min(mi, dfs(e.second, p) + e.first); } return d[p] = mi; } int main() { cin.tie(0); ios::sync_with_stdio(0); int n, q, c; cin >> n >> q >> c; edge = vector<vector<P>>(n); for (int i = 0; i < n - 1; ++i) { int u, v, l; cin >> u >> v >> l; --u; --v; edge[u].push_back({l, v}); edge[v].push_back({l, u}); } vector<int> x(q); for (int i = 0; i < q; ++i) { cin >> x[i]; --x[i]; } vector<vector<long long>> dist(n, vector<long long>(n, LM)); for (int i = 0; i < n; ++i) { queue<int> que; dist[i][i] = 0; que.push(i); while (!que.empty()) { int p = que.front(); que.pop(); for (auto& e : edge[p]) { if (dist[i][e.second] > dist[i][p] + e.first) { dist[i][e.second] = dist[i][p] + e.first; que.push(e.second); } } } } vector<long long> dp(n, LM); dp[x[0]] = 0; d = vector<long long>(n, LM); for (int i = 1; i < q; ++i) { for (int j = 0; j < n; ++j) { d[j] = dp[j] + c; } for (int j = 0; j < n; ++j) { d[x[i - 1]] = min(d[x[i - 1]], dp[j]); } dfs(x[i], -1); for (int j = 0; j < n; ++j) { dp[j] = min(dp[j] + dist[x[i - 1]][x[i]], d[j] + dist[j][x[i]]); } } long long ans = LM; for (int i = 0; i < n; ++i) { ans = min(ans, dp[i]); } cout << ans << '\n'; return 0; }