結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:36:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 89 ms / 2,000 ms |
コード長 | 1,067 bytes |
コンパイル時間 | 2,796 ms |
コンパイル使用メモリ | 181,336 KB |
最終ジャッジ日時 | 2025-01-19 10:43:23 |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <cstdio>#include <cstring>#include <iostream>#include <string>#include <cmath>#include <bitset>#include <vector>#include <map>#include <set>#include <queue>#include <deque>#include <algorithm>#include <complex>#include <unordered_map>#include <unordered_set>#include <random>#include <cassert>#include <fstream>#include <utility>#include <functional>#include <time.h>#include <stack>#include <array>#include <list>#include <atcoder/all>#define popcount __builtin_popcountusing namespace std;using namespace atcoder;typedef long long ll;typedef pair<int, int> P;ll n;vector<int> g[100010];ll cnt[100010];ll ans;void dfs(int x, int p){cnt[x]=1;for(auto y:g[x]){if(y==p) continue;dfs(y, x);cnt[x]+=cnt[y];ans+=cnt[y]*(n-cnt[y]);}ans+=cnt[x]*(n-cnt[x])+n;}int main(){cin>>n;for(int i=0; i<n-1; i++){int a, b; cin>>a>>b;a--; b--;g[a].push_back(b);g[b].push_back(a);}dfs(0, -1);cout<<ans<<endl;return 0;}