結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー snow39snow39
提出日時 2021-03-05 21:54:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 58 ms / 2,000 ms
コード長 1,277 bytes
コンパイル時間 1,056 ms
コンパイル使用メモリ 110,240 KB
最終ジャッジ日時 2025-01-19 10:56:17
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}
int main(){
cin.tie(0);
ios::sync_with_stdio(false);
int N;
cin>>N;
vector<vector<int>> g(N);
rep(i,N-1){
int a,b;
cin>>a>>b;
a--;b--;
g[a].push_back(b);
g[b].push_back(a);
}
vector<ll> sz(N,0);
auto szdfs = [&](auto &&szdfs,int now,int par)->void{
sz[now]=1;
for(auto nex:g[now]){
if(nex==par) continue;
szdfs(szdfs,nex,now);
sz[now]+=sz[nex];
}
};
szdfs(szdfs,0,-1);
ll ans=0;
auto dfs = [&](auto &&dfs,int now,int par)->void{
for(auto nex:g[now]){
if(nex==par) continue;
dfs(dfs,nex,now);
ll res=sz[nex]*(N-sz[nex]);
ans+=res;
}
ans+=N;
if(par!=-1){
ll res=(N-sz[now])*sz[now];
ans+=res;
}
};
dfs(dfs,0,-1);
cout<<ans<<endl;
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0