結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
|
提出日時 | 2021-03-05 22:35:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 163 ms / 2,000 ms |
コード長 | 1,617 bytes |
コンパイル時間 | 1,883 ms |
コンパイル使用メモリ | 196,216 KB |
最終ジャッジ日時 | 2025-01-19 11:32:38 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include<bits/stdc++.h>using namespace std;#define rep(i,n) for(ll i=0;i<n;i++)#define repl(i,l,r) for(ll i=(l);i<(r);i++)#define per(i,n) for(ll i=(n)-1;i>=0;i--)#define perl(i,r,l) for(ll i=r-1;i>=l;i--)#define fi first#define se second#define pb push_back#define ins insert#define pqueue(x) priority_queue<x,vector<x>,greater<x>>#define all(x) (x).begin(),(x).end()#define CST(x) cout<<fixed<<setprecision(x)#define vtpl(x,y,z) vector<tuple<x,y,z>>#define rev(x) reverse(x);using ll=long long;using vl=vector<ll>;using vvl=vector<vector<ll>>;using pl=pair<ll,ll>;using vpl=vector<pl>;using vvpl=vector<vpl>;const ll MOD=1000000007;const ll MOD9=998244353;const int inf=1e9+10;const ll INF=4e18;const ll dy[8]={1,0,-1,0,1,1,-1,-1};const ll dx[8]={0,-1,0,1,1,-1,1,-1};template<class T> inline bool chmin(T& a, T b) {if (a > b) {a = b;return true;}return false;}template<class T> inline bool chmax(T& a, T b) {if (a < b) {a = b;return true;}return false;}ll n;vvl g(100010);ll dp[100010];ll ans=0;void dfs(ll v,ll par=-1){ll ret=1;for(auto p:g[v]){if(p==par)continue;dfs(p,v);ret+=dp[p];}dp[v]=ret;}void dfs2(ll v,ll par=-1){ans+=n;ll rem=n-1;for(auto p:g[v]){if(p==par)continue;rem-=dp[p];ans+=dp[p]*(n-dp[p]);dfs2(p,v);}ans+=rem*(n-rem);}int main(){cin >> n;rep(i,n-1){ll a,b;cin >> a >> b;a--;b--;g[a].pb(b);g[b].pb(a);}dfs(0);dfs2(0);cout << ans <<endl;}