結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2022-02-19 18:10:03 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,366 bytes |
コンパイル時間 | 1,108 ms |
コンパイル使用メモリ | 90,872 KB |
実行使用メモリ | 15,616 KB |
最終ジャッジ日時 | 2024-06-29 10:35:20 |
合計ジャッジ時間 | 3,787 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 2 |
other | AC * 1 WA * 40 |
ソースコード
#include <iostream>#include <cstdio>#include <string>#include <algorithm>#include <utility>#include <cmath>#include <vector>#include <stack>#include <queue>#include <deque>#include <set>#include <map>#include <tuple>#include <numeric>#include <functional>using namespace std;typedef long long ll;typedef vector<ll> vl;typedef vector<vector<ll>> vvl;typedef pair<ll, ll> P;#define rep(i, n) for(ll i = 0; i < n; i++)#define exrep(i, a, b) for(ll i = a; i <= b; i++)#define out(x) cout << x << endl#define exout(x) printf("%.10f\n", x)#define chmax(x, y) x = max(x, y)#define chmin(x, y) x = min(x, y)#define all(a) a.begin(), a.end()#define rall(a) a.rbegin(), a.rend()#define pb push_back#define re0 return 0const ll mod = 998244353;const ll INF = 1e16;ll n;vvl G;vl sz; // sz[v] : vの部分木のサイズll ans = 0;void dfs(ll v, ll p = -1) {if(p != -1 && G[v].size() == 1) {ans += n;}for(ll u : G[v]) {if(u == p) { continue; }dfs(u, v);sz[v] += sz[u];}}int main() {cin >> n;G.resize(n);rep(i, n-1) {ll a, b;cin >> a >> b;a--; b--;G[a].pb(b);G[b].pb(a);}sz.assign(n, 1);dfs(0);rep(v, n) {ans += sz[v] * (2*n + 1 - 2*sz[v]);}out(ans);re0;}