結果
問題 |
No.1418 Sum of Sum of Subtree Size
|
ユーザー |
|
提出日時 | 2021-03-21 23:39:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 1,170 bytes |
コンパイル時間 | 4,154 ms |
コンパイル使用メモリ | 197,436 KB |
最終ジャッジ日時 | 2025-01-19 20:57:34 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include<bits/stdc++.h> using namespace std; typedef long long ll; typedef long double ld; #define REP(i, n) for (int i = 0; i < (n); ++i) #define REPR(i, n) for (int i = n - 1; i >= 0; --i) #define FOR(i, m, n) for (int i = m; i < n; ++i) #define FORR(i, m, n) for (int i = m; i >= n; --i) #define ALL(v) (v).begin(),(v).end() template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; } const ll INF=1LL<<60; const int inf=(1<<30)-1; const int mod=1e9+7; int dx[8]={1,0,-1,0,-1,-1,1,1}; int dy[8]={0,1,0,-1,-1,1,-1,1}; const int nmx=100005; vector<vector<int>> e(nmx); vector<ll> c(nmx); ll ans=0; ll n; void dfs(int cur,int pre=-1){ for(int to:e[cur]){ if(to==pre) continue; dfs(to,cur); ans+=c[to]*(n-c[to]); c[cur]+=c[to]; } c[cur]++; ans+=c[cur]*(n-c[cur]); } int main(){ cin.tie(0); ios::sync_with_stdio(false); cin >> n; REP(i,n-1){ int a,b;cin >> a >> b; a--,b--; e[a].push_back(b); e[b].push_back(a); } dfs(0); cout << ans+n*n << endl; }