結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-12 00:03:28 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,929 bytes |
コンパイル時間 | 2,140 ms |
コンパイル使用メモリ | 182,196 KB |
実行使用メモリ | 12,416 KB |
最終ジャッジ日時 | 2024-10-13 10:56:00 |
合計ジャッジ時間 | 5,399 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 WA * 6 |
ソースコード
#pragma GCC optimize("O3")//#pragma GCC target("avx2")//#pragma GCC optimize("unroll-loops")#include <bits/stdc++.h>using namespace std;typedef long long ll;typedef unsigned long long ull;typedef long double ld;typedef vector<int> vi;typedef vector<ll> vll;typedef vector<ld> vld;typedef pair<ll,ll> Pll;typedef pair<int,int> Pin;ll INF = 1e16;int inf = 1e9;#define ALL(x) (x).begin(), (x).end()#define FOR(i, m, n) for (ll i = (m); i < (n); ++i)#define REVFOR(i, m, n) for (ll i = (n - 1); i >= (m); --i)#define REP(i, n) FOR(i, 0, n)#define REVREP(i, n) REVFOR(i, 0, n)#define fi first#define se second#define pb push_back#define mp make_pair#define eb emplace_back#define bcnt __builtin_popcountll#ifdef LOCAL#include <prettyprint.hpp>#define debug(...) cerr << "[" << #__VA_ARGS__ << "]: ", d_err(__VA_ARGS__);#else#define debug(...) 83;#endifvoid d_err() {cerr << endl;}template <typename H, typename... T>void d_err(H h, T... t) {cerr << h << " ";d_err(t...);}template <typename T>void print(T x) {cout << x << "\n";}template <typename T>void print(vector<T>& x) {int N = x.size();REP(i, N) {if (i > 0) cout << " ";cout << x[i];}cout << "\n";}int main(){cin.tie(0);ios_base::sync_with_stdio(false);cout << fixed << setprecision(20);int n; cin >> n;vector<vi> edge(n);REP(i, n-1) {int a, b; cin >> a >> b;a--, b--;edge[a].eb(b);edge[b].eb(a);}vi dp(n);ll ans = 0;auto dfs = [&](auto &&f, int cur, int par = -1) -> void {dp[cur]++;for(auto &e: edge[cur]) {if (e == par) continue;f(f, e, cur);dp[cur] += dp[e];ans += dp[e] * (n - dp[e]);}ans += (n - dp[cur]) * dp[cur];ans += n;};dfs(dfs, 0);print(ans);}