結果

問題 No.1098 LCAs
ユーザー betrue12
提出日時 2020-06-26 21:41:28
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 482 ms / 2,000 ms
コード長 644 bytes
コンパイル時間 2,020 ms
コンパイル使用メモリ 199,968 KB
最終ジャッジ日時 2025-01-11 11:06:24
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int main(){
    int N;
    cin >> N;
    vector<int> edges[200000];
    for(int i=0; i<N-1; i++){
        int a, b;
        cin >> a >> b;
        a--; b--;
        edges[a].push_back(b);
        edges[b].push_back(a);
    }

    vector<int64_t> sum(N), ans(N);
    auto dfs = [&](auto&& dfs, int i, int p)->void{
        for(int j : edges[i]) if(j != p){
            dfs(dfs, j, i);
            sum[i] += sum[j];
            ans[i] -= sum[j]*sum[j];
        }
        sum[i]++;
        ans[i] += sum[i]*sum[i];
    };
    dfs(dfs, 0, -1);
    for(auto a : ans) cout << a << endl;
    return 0;
}
0