結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー | fumofumofuni |
提出日時 | 2021-03-05 22:35:58 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 108 ms / 2,000 ms |
コード長 | 1,617 bytes |
コンパイル時間 | 2,193 ms |
コンパイル使用メモリ | 199,272 KB |
実行使用メモリ | 17,152 KB |
最終ジャッジ日時 | 2024-04-16 10:21:36 |
合計ジャッジ時間 | 5,036 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 3 ms
5,504 KB |
testcase_01 | AC | 3 ms
5,632 KB |
testcase_02 | AC | 4 ms
5,504 KB |
testcase_03 | AC | 100 ms
9,984 KB |
testcase_04 | AC | 100 ms
9,984 KB |
testcase_05 | AC | 103 ms
9,984 KB |
testcase_06 | AC | 102 ms
9,984 KB |
testcase_07 | AC | 100 ms
9,984 KB |
testcase_08 | AC | 64 ms
8,704 KB |
testcase_09 | AC | 29 ms
6,912 KB |
testcase_10 | AC | 32 ms
7,040 KB |
testcase_11 | AC | 17 ms
6,400 KB |
testcase_12 | AC | 50 ms
7,936 KB |
testcase_13 | AC | 59 ms
8,576 KB |
testcase_14 | AC | 64 ms
8,576 KB |
testcase_15 | AC | 48 ms
7,808 KB |
testcase_16 | AC | 8 ms
5,888 KB |
testcase_17 | AC | 8 ms
5,888 KB |
testcase_18 | AC | 93 ms
9,728 KB |
testcase_19 | AC | 16 ms
6,272 KB |
testcase_20 | AC | 5 ms
5,632 KB |
testcase_21 | AC | 44 ms
7,680 KB |
testcase_22 | AC | 39 ms
7,424 KB |
testcase_23 | AC | 6 ms
5,760 KB |
testcase_24 | AC | 6 ms
5,760 KB |
testcase_25 | AC | 5 ms
5,760 KB |
testcase_26 | AC | 6 ms
6,016 KB |
testcase_27 | AC | 4 ms
5,760 KB |
testcase_28 | AC | 4 ms
5,632 KB |
testcase_29 | AC | 6 ms
6,016 KB |
testcase_30 | AC | 6 ms
5,760 KB |
testcase_31 | AC | 5 ms
5,632 KB |
testcase_32 | AC | 5 ms
5,632 KB |
testcase_33 | AC | 22 ms
8,192 KB |
testcase_34 | AC | 108 ms
17,152 KB |
testcase_35 | AC | 41 ms
10,496 KB |
testcase_36 | AC | 10 ms
6,400 KB |
testcase_37 | AC | 62 ms
9,976 KB |
testcase_38 | AC | 56 ms
9,724 KB |
testcase_39 | AC | 3 ms
5,504 KB |
testcase_40 | AC | 4 ms
5,504 KB |
testcase_41 | AC | 4 ms
5,504 KB |
testcase_42 | AC | 4 ms
5,504 KB |
testcase_43 | AC | 3 ms
5,632 KB |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define fi first #define se second #define pb push_back #define ins insert #define pqueue(x) priority_queue<x,vector<x>,greater<x>> #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define vtpl(x,y,z) vector<tuple<x,y,z>> #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=1e9+10; const ll INF=4e18; const ll dy[8]={1,0,-1,0,1,1,-1,-1}; const ll dx[8]={0,-1,0,1,1,-1,1,-1}; template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } ll n; vvl g(100010); ll dp[100010]; ll ans=0; void dfs(ll v,ll par=-1){ ll ret=1; for(auto p:g[v]){ if(p==par)continue; dfs(p,v); ret+=dp[p]; } dp[v]=ret; } void dfs2(ll v,ll par=-1){ ans+=n; ll rem=n-1; for(auto p:g[v]){ if(p==par)continue; rem-=dp[p]; ans+=dp[p]*(n-dp[p]); dfs2(p,v); } ans+=rem*(n-rem); } int 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);dfs2(0); cout << ans <<endl; }