結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2020-07-05 00:58:15 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 554 ms / 2,000 ms |
コード長 | 1,063 bytes |
コンパイル時間 | 1,893 ms |
コンパイル使用メモリ | 174,872 KB |
実行使用メモリ | 31,916 KB |
最終ジャッジ日時 | 2024-09-19 18:11:06 |
合計ジャッジ時間 | 10,090 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#include <bits/stdc++.h>#define all(x) (x).begin(), (x).end()typedef long long ll;#define MOD 1000000007using namespace std;vector<ll> num;vector<ll> ans;vector<vector<int>> edge;int dfs(int now, int prev) {int sum = 0;for(auto next : edge[now]) {if(next != prev) {int tmp = dfs(next, now);num[next] = tmp;sum += tmp;}}return sum + 1;}ll solve(int now, int prev) {ll res = num[now] * num[now];ll sub = 0;for(auto next : edge[now]) {if(next != prev) {ll tmp = solve(next, now);sub += tmp;}}ans[now] = res - sub;return res;}int main() {int n;cin >> n;edge.resize(n);for(int i = 0; i < n - 1; i++) {int v, w;cin >> v >> w;edge[v - 1].push_back(w - 1);edge[w - 1].push_back(v - 1);}num.resize(n, -1);ans.resize(n, 0);num[0] = dfs(0, -1);solve(0, -1);for(int i = 0; i < n; i++) {cout << ans[i] << endl;}}