結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-07 13:38:29 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 57 ms / 2,000 ms |
コード長 | 1,540 bytes |
コンパイル時間 | 1,996 ms |
コンパイル使用メモリ | 198,336 KB |
最終ジャッジ日時 | 2025-01-19 12:35:33 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include<bits/stdc++.h>using namespace std;#define ALL(x) begin(x),end(x)#define rep(i,n) for(int i=0;i<(n);i++)#define debug(v) cout<<#v<<":";for(auto x:v){cout<<x<<' ';}cout<<endl;#define mod 1000000007using ll=long long;const int INF=1000000000;const ll LINF=1001002003004005006ll;int dx[]={1,0,-1,0},dy[]={0,1,0,-1};// ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}struct IOSetup{IOSetup(){cin.tie(0);ios::sync_with_stdio(0);cout<<fixed<<setprecision(12);}} iosetup;template<typename T>ostream &operator<<(ostream &os,const vector<T>&v){for(int i=0;i<(int)v.size();i++) os<<v[i]<<(i+1==(int)v.size()?"":" ");return os;}template<typename T>istream &operator>>(istream &is,vector<T>&v){for(T &x:v)is>>x;return is;}#define int llsigned main(){int n;cin>>n;vector<vector<int>> g(n);rep(i,n-1){int u,v;cin>>u>>v;u--,v--;g[u].push_back(v);g[v].push_back(u);}ll res=0;vector<int> c(n,0);function<void(int,int)> dfs=[&](int pre,int cur){ll ret=0;for(auto to:g[cur])if(pre!=to){dfs(cur,to);ret+=c[to];}c[cur]=ret+1;for(auto to:g[cur])if(pre!=to) res+=(n-c[to])*c[to];res+=c[cur]*(n-c[cur]);};dfs(-1,0);cout<<res+n*n<<endl;return 0;}