結果

問題 No.1098 LCAs
コンテスト
ユーザー snow39
提出日時 2020-06-26 22:03:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 199 ms / 2,000 ms
コード長 1,079 bytes
コンパイル時間 1,057 ms
コンパイル使用メモリ 101,108 KB
実行使用メモリ 39,016 KB
最終ジャッジ日時 2024-07-04 21:11:57
合計ジャッジ時間 4,539 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}

vector<vector<int>> g;
vector<ll> ans;

ll dfs(int now,int par){
  ll sum=0;
  ll res=0;
  vector<ll> v;
  for(auto nex:g[now]){
    if(nex==par) continue;
    ll val=dfs(nex,now);
    sum+=val;
    v.push_back(val);
  }

  for(auto p:v){
    res+=p*(sum-p);
  }
  res+=2*sum+1;
  ans[now]=res;
  return sum+1;
}

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N;
  cin>>N;
  g.resize(N);
  rep(i,N-1){
    int a,b;
    cin>>a>>b;
    a--;b--;
    g[a].push_back(b);
    g[b].push_back(a);
  }

  ans.resize(N,0);

  dfs(0,-1);

  rep(i,N) cout<<ans[i]<<"\n";



  return 0;
}
0