#504493 (C++17) No.1098 LCAs

• 問題: No.1098 LCAs
• ユーザー: fura
• 提出日時: 2020-06-28 16:37:09
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 709 bytes
• コンパイル時間: 2,074 ms
• コンパイル使用メモリ: 208,396 KB
• 実行使用メモリ: 36,936 KB
• 最終ジャッジ日時: 2024-07-07 20:21:04
• 合計ジャッジ時間: 5,513 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,812 KB
testcase_01	AC	2 ms; 6,944 KB
testcase_02	AC	2 ms; 6,944 KB
testcase_03	AC	1 ms; 6,944 KB
testcase_04	AC	1 ms; 6,940 KB
testcase_05	AC	1 ms; 6,940 KB
testcase_06	AC	1 ms; 6,940 KB
testcase_07	AC	2 ms; 6,940 KB
testcase_08	AC	1 ms; 6,940 KB
testcase_09	AC	1 ms; 6,940 KB
testcase_10	AC	1 ms; 6,944 KB
testcase_11	AC	1 ms; 6,940 KB
testcase_12	AC	2 ms; 6,940 KB
testcase_13	AC	2 ms; 6,940 KB
testcase_14	AC	2 ms; 6,944 KB
testcase_15	AC	1 ms; 6,940 KB
testcase_16	AC	2 ms; 6,944 KB
testcase_17	AC	2 ms; 6,940 KB
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-

ソースコード

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;
using lint=long long;

using graph=vector<vector<int>>;

void add_undirected_edge(graph& G,int u,int v){
	G[u].emplace_back(v);
	G[v].emplace_back(u);
}

graph T;
int sz[200000];
lint ans[200000];

void dfs(int u,int p){
	sz[u]=1;

	vector<lint> seq;
	for(int v:T[u]) if(v!=p) {
		dfs(v,u);
		sz[u]+=sz[v];
		seq.emplace_back(sz[v]);
	}

	ans[u]=(sz[u]-1)*(sz[u]-1);
	for(auto k:seq) ans[u]-=k*k;
	ans[u]+=2*sz[u]-1;
}

int main(){
	int n; scanf("%d",&n);
	T.resize(n);
	rep(i,n-1){
		int u,v; scanf("%d%d",&u,&v); u--; v--;
		add_undirected_edge(T,u,v);
	}

	dfs(0,-1);

	rep(u,n) printf("%lld\n",ans[u]);

	return 0;
}