結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:54:24 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 58 ms / 2,000 ms |
コード長 | 1,277 bytes |
コンパイル時間 | 1,056 ms |
コンパイル使用メモリ | 110,240 KB |
最終ジャッジ日時 | 2025-01-19 10:56:17 |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#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;}int main(){cin.tie(0);ios::sync_with_stdio(false);int N;cin>>N;vector<vector<int>> g(N);rep(i,N-1){int a,b;cin>>a>>b;a--;b--;g[a].push_back(b);g[b].push_back(a);}vector<ll> sz(N,0);auto szdfs = [&](auto &&szdfs,int now,int par)->void{sz[now]=1;for(auto nex:g[now]){if(nex==par) continue;szdfs(szdfs,nex,now);sz[now]+=sz[nex];}};szdfs(szdfs,0,-1);ll ans=0;auto dfs = [&](auto &&dfs,int now,int par)->void{for(auto nex:g[now]){if(nex==par) continue;dfs(dfs,nex,now);ll res=sz[nex]*(N-sz[nex]);ans+=res;}ans+=N;if(par!=-1){ll res=(N-sz[now])*sz[now];ans+=res;}};dfs(dfs,0,-1);cout<<ans<<endl;return 0;}