#493982 (C++17) No.277 根掘り葉掘り

• 問題: No.277 根掘り葉掘り
• ユーザー: fura
• 提出日時: 2020-06-08 08:52:57
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 66 ms / 3,000 ms
• コード長: 821 bytes
• コンパイル時間: 2,129 ms
• コンパイル使用メモリ: 207,068 KB
• 実行使用メモリ: 14,056 KB
• 最終ジャッジ日時: 2023-08-26 00:52:25
• 合計ジャッジ時間: 4,236 ms
• ジャッジサーバーID （参考情報）: judge15 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,356 KB
testcase_01	AC	2 ms; 4,356 KB
testcase_02	AC	2 ms; 4,352 KB
testcase_03	AC	2 ms; 4,356 KB
testcase_04	AC	2 ms; 4,356 KB
testcase_05	AC	1 ms; 4,352 KB
testcase_06	AC	2 ms; 4,360 KB
testcase_07	AC	1 ms; 4,352 KB
testcase_08	AC	1 ms; 4,356 KB
testcase_09	AC	66 ms; 14,056 KB
testcase_10	AC	46 ms; 10,312 KB
testcase_11	AC	57 ms; 9,724 KB
testcase_12	AC	64 ms; 12,024 KB
testcase_13	AC	64 ms; 9,868 KB
testcase_14	AC	58 ms; 10,204 KB
testcase_15	AC	61 ms; 10,016 KB
testcase_16	AC	62 ms; 9,912 KB
testcase_17	AC	60 ms; 9,900 KB
testcase_18	AC	63 ms; 9,624 KB
testcase_19	AC	60 ms; 9,864 KB

ソースコード

#include <bits/stdc++.h>

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

using namespace std;

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);
}

const int INF=1<<29;

graph T;
int dep[100000];

void dfs(int u,int p){
	for(int v:T[u]) if(v!=p) {
		dep[v]=dep[u]+1;
		dfs(v,u);
	}
}

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);

	int d[100000];
	queue<int> Q;
	rep(u,n){
		if(u!=0 && T[u].size()==1){
			d[u]=0;
			Q.emplace(u);
		}
		else{
			d[u]=INF;
		}
	}
	while(!Q.empty()){
		int u=Q.front(); Q.pop();
		for(int v:T[u]) if(d[v]==INF) {
			d[v]=d[u]+1;
			Q.emplace(v);
		}
	}

	rep(u,n) printf("%d\n",min(dep[u],d[u]));

	return 0;
}