結果
問題 | 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 doubletypedef 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;}