結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 22:01:41 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 45 ms / 2,000 ms |
コード長 | 1,479 bytes |
コンパイル時間 | 1,745 ms |
コンパイル使用メモリ | 171,592 KB |
実行使用メモリ | 17,152 KB |
最終ジャッジ日時 | 2024-10-07 01:54:27 |
合計ジャッジ時間 | 3,358 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h>//#include <chrono>//#pragma GCC optimize("Ofast")using namespace std;#define reps(i,s,n) for(int i = s; i < n; i++)#define rep(i,n) reps(i,0,n)#define Rreps(i,n,e) for(int i = n - 1; i >= e; --i)#define Rrep(i,n) Rreps(i,n,0)#define ALL(a) a.begin(), a.end()using ll = long long;using vec = vector<ll>;using mat = vector<vec>;ll N,M,H,W,Q,K,A,B;string S;using P = pair<ll, ll>;const ll INF = (1LL<<60);template<class T> bool chmin(T &a, const T b){if(a > b) {a = b; return true;}else return false;}template<class T> bool chmax(T &a, const T b){if(a < b) {a = b; return true;}else return false;}template<class T> void my_printv(std::vector<T> v,bool endline = true){if(!v.empty()){for(std::size_t i{}; i<v.size()-1; ++i) std::cout<<v[i]<<" ";std::cout<<v.back();}if(endline) std::cout<<std::endl;}ll res;vec sz;void dfs(int p, int v, mat &G){sz[v] = 1;for(int to : G[v]){if(to != p){dfs(v, to, G);res += sz[to] * (N - sz[to]);sz[v] += sz[to];}}if(p != -1){res += (N - sz[v]) * sz[v];}}int main(){cin.tie(nullptr);ios::sync_with_stdio(false);cin>>N;mat G(N);rep(i, N - 1){int a, b;cin>>a>>b; --a; --b;G[a].push_back(b);G[b].push_back(a);}res = N * N;sz.resize(N);dfs(-1, 0, G);cout<<res<<endl;}