結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
![]() |
提出日時 | 2021-03-06 00:02:11 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 92 ms / 2,000 ms |
コード長 | 1,641 bytes |
コンパイル時間 | 2,045 ms |
コンパイル使用メモリ | 196,088 KB |
最終ジャッジ日時 | 2025-01-19 12:07:19 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; using ull = unsigned long long; #define v(t) vector<t> #define p(t) pair<t, t> #define p2(t, s) pair<t, s> #define vp(t) v(p(t)) #define rep(i, n) for (int i=0,i##_len=((int)(n)); i<i##_len; ++i) #define rep2(i, a, n) for (int i=((int)(a)),i##_len=((int)(n)); i<=i##_len; ++i) #define repr(i, n) for (int i=((int)(n)-1); i>=0; --i) #define rep2r(i, a, n) for (int i=((int)(n)),i##_len=((int)(a)); i>=i##_len; --i) #define repi(itr, c) for (__typeof((c).begin()) itr=(c).begin(); itr!=(c).end(); ++itr) #define repir(itr, c) for (__typeof((c).rbegin()) itr=(c).rbegin(); itr!=(c).rend(); ++itr) #define sz(x) ((int)(x).size()) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define SORT(v, n) sort(v, v+n); #define VSORT(v) sort(v.begin(), v.end()); #define RSORT(x) sort(rall(x)); #define pb push_back #define eb emplace_back #define INF (1e9) #define LINF (1e18) #define PI (acos(-1)) #define EPS (1e-7) #define DEPS (1e-10) int n; ll ans = 0; int dfs(v(v(int))& g, int vi, int pi=-1) { int sum = 1; rep(i, sz(g[vi])) { if (g[vi][i] == pi) continue; int res = dfs(g, g[vi][i], vi); ans += (ll)res * (n - res); sum += res; } if (pi != -1) ans += (ll)(n-sum) * sum; ans += n; return sum; } int main(){ cin >> n; v(v(int)) g(n); rep(i, n-1) { int a, b; cin >> a >> b; a--; b--; g[a].pb(b); g[b].pb(a); } dfs(g, 0); cout << ans << endl; return 0; }