結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
![]() |
提出日時 | 2021-03-06 17:09:47 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 95 ms / 2,000 ms |
コード長 | 1,040 bytes |
コンパイル時間 | 1,760 ms |
コンパイル使用メモリ | 171,304 KB |
実行使用メモリ | 15,360 KB |
最終ジャッジ日時 | 2024-10-08 17:15:15 |
合計ジャッジ時間 | 5,287 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define int long long #define REP(i,m,n) for(int i=(m);i<(n);i++) #define rep(i,n) REP(i,0,n) #define pb push_back #define all(a) a.begin(),a.end() #define rall(c) (c).rbegin(),(c).rend() #define mp make_pair #define endl '\n' #define vec vector<ll> #define mat vector<vector<ll> > #define fi first #define se second #define double long double typedef long long ll; typedef unsigned long long ull; typedef pair<ll,ll> pll; //typedef long double ld; typedef complex<double> Complex; const ll INF=1e9+7; const ll MOD=998244353; const ll inf=INF*INF; const ll mod=INF; const ll MAX=140010; vector<vector<ll> >G(MAX); vector<ll>sz(MAX); ll n; ll ans=0; ll dfs(ll i,ll p){ sz[i]=1; for(auto e:G[i]){ if(e==p)continue; sz[i]+=dfs(e,i); } ans+=sz[i]*(n-sz[i])*2LL; return sz[i]; } signed main(){ cin>>n; rep(i,n-1){ ll a,b;cin>>a>>b; a--;b--; G[a].pb(b); G[b].pb(a); } dfs(0,-1); ans+=n*n; cout<<ans<<endl; }