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

• 問題: No.277 根掘り葉掘り
• ユーザー: fura
• 提出日時: 2020-06-08 08:52:57
• 言語: C++17 (gcc 11.2.0 + boost 1.78.0)
• 結果: AC
• 実行時間: 63 ms / 3,000 ms
• コード長: 821 bytes
• コンパイル時間: 2,181 ms
• 使用メモリ: 13,980 KB
• 最終ジャッジ日時: 2023-01-13 18:39:32
• 合計ジャッジ時間: 4,334 ms
• ジャッジサーバーID （参考情報）: judge12 / judge13

入力	結果	実行時間; 使用メモリ
testcase_00	AC	2 ms; 3,664 KB
testcase_01	AC	2 ms; 3,568 KB
testcase_02	AC	2 ms; 3,588 KB
testcase_03	AC	2 ms; 3,612 KB
testcase_04	AC	2 ms; 3,576 KB
testcase_05	AC	1 ms; 3,604 KB
testcase_06	AC	2 ms; 3,688 KB
testcase_07	AC	1 ms; 3,688 KB
testcase_08	AC	2 ms; 3,652 KB
testcase_09	AC	63 ms; 13,980 KB
testcase_10	AC	43 ms; 10,256 KB
testcase_11	AC	52 ms; 9,748 KB
testcase_12	AC	60 ms; 11,792 KB
testcase_13	AC	58 ms; 9,920 KB
testcase_14	AC	56 ms; 9,968 KB
testcase_15	AC	57 ms; 9,964 KB
testcase_16	AC	57 ms; 10,048 KB
testcase_17	AC	57 ms; 9,968 KB
testcase_18	AC	52 ms; 9,756 KB
testcase_19	AC	57 ms; 9,924 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;
}