結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
|
提出日時 | 2021-03-05 21:54:00 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 49 ms / 2,000 ms |
コード長 | 2,472 bytes |
コンパイル時間 | 1,795 ms |
コンパイル使用メモリ | 200,504 KB |
最終ジャッジ日時 | 2025-01-19 10:56:10 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define REP(i, n) for(int i=0; i<(n); ++i)#define RREP(i, n) for(int i=(n);i>=0;--i)#define FOR(i, a, n) for (int i=(a); i<(n); ++i)#define RFOR(i, a, b) for(int i=(a);i>=(b);--i)#define SZ(x) ((int)(x).size())#define ALL(x) (x).begin(),(x).end()#define DUMP(x) cerr<<#x<<" = "<<(x)<<endl#define DEBUG(x) cerr<<#x<<" = "<<(x)<<" (L"<<__LINE__<<")"<<endl;template<class T>ostream &operator<<(ostream &os, const vector<T> &v) {REP(i, SZ(v)) {if (i) os << " ";os << v[i];}return os;}template <class T>void debug(const vector<T> &v) {cout << "[";REP(i, SZ(v)) {if(i) cout << ", ";cout << v[i];}cout << "]" << endl;}template<class T, class U>ostream &operator<<(ostream &os, const pair<T, U> &p) {return os << p.first << " " << p.second;}template <class T, class U>void debug(const pair<T, U> &p) {cout << "(" << p.first << " " << p.second << ")" << endl;}template<class T>bool chmax(T &a, const T &b) {if (a < b) {a = b;return true;}return false;}template<class T>bool chmin(T &a, const T &b) {if (b < a) {a = b;return true;}return false;}using ll = long long;using ull = unsigned long long;using ld = long double;using P = pair<int, int>;using vi = vector<int>;using vll = vector<ll>;using vvi = vector<vi>;using vvll = vector<vll>;const ll MOD = 1e9 + 7;const ll MOD998 = 998244353;const int INF = INT_MAX;const ll LINF = LLONG_MAX;const int inf = INT_MIN;const ll linf = LLONG_MIN;const ld eps = 1e-9;int main() {cin.tie(0);ios::sync_with_stdio(false);cout << fixed << setprecision(10);int n; cin >> n;vvi g(n);REP(i, n-1) {int a, b; cin >> a >> b;a--; b--;g[a].push_back(b);g[b].push_back(a);}vll sz(n);function<void(int, int)> dfs = [&](int now, int prev) {sz[now] = 1;for(auto &nxt: g[now]) {if(nxt == prev) continue;dfs(nxt, now);sz[now] += sz[nxt];}};dfs(0, -1);ll ans = 0;function<void(int, int)> dfs2 = [&](int now, int prev) {ans += n + sz[now] * (n - sz[now]);for(auto &nxt: g[now]) {if(nxt == prev) continue;dfs2(nxt, now);ans += sz[nxt] * (n - sz[nxt]);}};dfs2(0, -1);cout << ans << endl;return 0;}