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