結果

問題 No.1098 LCAs
ユーザー akun0716akun0716
提出日時 2020-06-26 22:20:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 230 ms / 2,000 ms
コード長 2,074 bytes
コンパイル時間 832 ms
コンパイル使用メモリ 85,216 KB
実行使用メモリ 30,664 KB
最終ジャッジ日時 2024-07-04 22:03:20
合計ジャッジ時間 4,959 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include<vector>
#include<algorithm>
#include<string>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<math.h>
using namespace std;
typedef long long ll;
#define int long long
#define double long double
typedef vector<int> VI;
typedef pair<int, int> pii;
typedef vector<pii> VP;
typedef vector<string> VS;
typedef priority_queue<int> PQ;
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; }
#define fore(i,a) for(auto &i:a)
#define REP(i,n) for(int i=0;i<n;i++)
#define eREP(i,n) for(int i=0;i<=n;i++)
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define eFOR(i,a,b) for(int i=(a);i<=(b);++i)
#define SORT(c) sort((c).begin(),(c).end())
#define rSORT(c) sort((c).rbegin(),(c).rend())
#define LB(x,a) lower_bound((x).begin(),(x).end(),(a))
#define UB(x,a) upper_bound((x).begin(),(x).end(),(a))
#define INF 1000000000
#define LLINF 9223372036854775807
#define mod 100000000000007
#define eps 1e-12 
//priority_queue<int,vector<int>, greater<int> > q2;
int comb(int a, int b) {
	if (a < b)return 0;
	if (b > a - b) return comb(a, a - b);


	int ansMul = 1;
	int ansDiv = 1;
	for (int i = 0; i < b; i++)
	{
		ansMul *= (a - i);
		ansDiv *= (i + 1);
		
	}
	return ansMul / ansDiv;
}
int N;
VI G[200010];
int ans[200010];
int t[200010];//size of subtree
int dp[200010];

void dfs1(int now, int pa) {
	fore(to, G[now]) {
		if (to == pa)continue;
		dfs1(to, now);
	}
	t[now]++;
	if (pa != -1)t[pa] += t[now];
	return;
}
void dfs(int now, int pa) {
	int cnt = comb(t[now], 2) * 2 + t[now];
	fore(to, G[now]) {
		if (to == pa)continue;
		dfs(to, now);
		cnt -= dp[to];
		dp[now] += dp[to];
	}
	//cout << now << " " << cnt << endl;
	ans[now] = cnt;
	dp[now] += cnt;
}

signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);

	cin >> N;
	REP(i, N - 1) {
		int a, b; cin >> a >> b; a--; b--;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	dfs1(0, -1);
	dfs(0, -1);
	REP(i, N) cout <<ans[i] << '\n';
	return 0;
}
0