結果
問題 |
No.1098 LCAs
|
ユーザー |
![]() |
提出日時 | 2020-06-26 23:02:50 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 344 ms / 2,000 ms |
コード長 | 912 bytes |
コンパイル時間 | 1,668 ms |
コンパイル使用メモリ | 174,456 KB |
実行使用メモリ | 49,920 KB |
最終ジャッジ日時 | 2024-07-04 23:25:06 |
合計ジャッジ時間 | 7,054 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#include<bits/stdc++.h> #define REP(i,n) for(int i=0,i##_len=int(n);i<i##_len;++i) #define rep(i,a,b) for(int i=int(a);i<int(b);++i) #define All(x) (x).begin(),(x).end() #define rAll(x) (x).rbegin(),(x).rend() using namespace std; using ll = long long; int main(){ int N; cin>>N; vector<vector<int>> graph(N); REP(i,N-1){ int a,b; cin>>a>>b; a--;b--; graph[a].push_back(b); graph[b].push_back(a); } vector<ll> ch(N); vector<vector<int>> chs(N); function<ll(int,int)> dfs = [&](int u,int pre){ ll res = 1; for(auto v:graph[u]){ if(v == pre) continue; res += dfs(v,u); chs[u].push_back(v); } return ch[u] = res; }; dfs(0,-1); REP(i,N){ ll ans = ch[i]*ch[i]; for(auto v:chs[i]){ ans -= ch[v]*ch[v]; } cout<<ans<<"\n"; } }