結果
問題 | No.277 根掘り葉掘り |
ユーザー |
|
提出日時 | 2024-09-12 15:55:56 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 248 ms / 3,000 ms |
コード長 | 1,581 bytes |
コンパイル時間 | 4,159 ms |
コンパイル使用メモリ | 285,556 KB |
実行使用メモリ | 15,840 KB |
最終ジャッジ日時 | 2024-09-12 15:56:05 |
合計ジャッジ時間 | 8,107 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include<bits/stdc++.h>using namespace std;const int INF = 1e9;int main() {int N;cin >> N;vector<vector<int>> E(N);for (int i = 0; i < N - 1; i++) {int A, B;cin >> A >> B;A--;B--;E[A].push_back(B);E[B].push_back(A);}vector<vector<int>> c(N);vector<int> p(N, -1);vector<int> bfs;queue<int> Q;Q.push(0);while (!Q.empty()) {int v = Q.front();Q.pop();bfs.push_back(v);for (int w : E[v]) {if (w != p[v]) {p[w] = v;c[v].push_back(w);Q.push(w);}}}reverse(bfs.begin(), bfs.end());vector<int> dp1(N, INF);//部分木に関しての最小のLfor (int v : bfs) {if (c[v].size() == 0) {dp1[v] = 0;}for (int w : c[v]) {dp1[v] = min(dp1[v], dp1[w] + 1);}}reverse(bfs.begin(), bfs.end());vector<int> dp2(N, INF);//0も葉とみなして辺を張り替えた後の葉との最短距離dp2[0] = 0;for (int v : bfs) {int deg = c[v].size();vector<int> left(deg + 1, INF);vector<int> right(deg + 2, INF);for (int i = 0; i < deg; i++) {int w = c[v][i];left[i + 1] = min(left[i], dp1[w]);}for (int i = deg - 1; i >= 0; i--) {int w = c[v][i];right[i] = min(right[i + 1], dp1[w]);}for (int i = 0; i < deg; i++) {int w = c[v][i];int tmp = min(left[i], right[i + 1]);tmp++;tmp = min(tmp, dp2[v]);tmp++;dp2[w] = tmp;}}for (int i = 0; i < N; i++) {cout << min(dp1[i], dp2[i]) << endl;}}