結果
問題 | No.2427 Tree Distance Two |
ユーザー |
|
提出日時 | 2023-08-22 19:49:48 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 753 ms / 2,000 ms |
コード長 | 813 bytes |
コンパイル時間 | 1,949 ms |
コンパイル使用メモリ | 174,688 KB |
実行使用メモリ | 35,976 KB |
最終ジャッジ日時 | 2024-12-17 20:45:42 |
合計ジャッジ時間 | 17,140 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 35 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll = long long;int main() {int N; cin >> N;vector<vector<int>> T(N);for(int i = 0; i < N-1; i++) {int a,b; cin >> a >> b;a--,b--;T[a].push_back(b);T[b].push_back(a);}vector<int> par(N,-1), cnt(N);auto dfs = [&](auto dfs, int i, int p) -> void{par[i] = p;for(auto ni : T[i]) {if(ni == p) continue;cnt[i]++;dfs(dfs,ni,i);}};dfs(dfs,0,-1);for(int i = 0; i < N; i++) {int ans = 0;if(par[i] != -1 && par[par[i]] != -1) ans++;for(int ni : T[i]) {if(par[ni] == i) ans += cnt[ni];}if(par[i] != -1) ans += cnt[par[i]] - 1;cout << ans << endl;}}