結果

問題 No.1098 LCAs
コンテスト
ユーザー QDSN
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define all(x) (x).begin(), (x).end()
typedef long long ll;
#define MOD 1000000007
using 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;
    }
}
0