結果

問題 No.277 根掘り葉掘り
ユーザー uafr_cs
提出日時 2015-09-04 22:54:28
言語 Java
(openjdk 23)
結果
AC  
実行時間 2,332 ms / 3,000 ms
コード長 1,902 bytes
コンパイル時間 2,359 ms
コンパイル使用メモリ 79,544 KB
実行使用メモリ 114,548 KB
最終ジャッジ日時 2024-07-19 01:30:57
合計ジャッジ時間 27,170 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Set;

public class Main {
	
	public static int INF = Integer.MAX_VALUE / 2 - 1;
	
	public static int[] bfs(final int N, ArrayList<LinkedHashSet<Integer>> adj, Set<Integer> starts){
		int[] costs = new int[N];
		Arrays.fill(costs, INF);
		
		
		LinkedList<Integer> node_queue = new LinkedList<Integer>();
		boolean[] in_queue = new boolean[N];
		for(final int start : starts){
			costs[start] = 0;
			node_queue.add(start);
			in_queue[start] = true;
		}
		
		
		while(!node_queue.isEmpty()){
			final int node = node_queue.poll();
			in_queue[node] = false;
			
			for(final int next : adj.get(node)){
				final int next_cost = costs[node] + 1;
				
				if(next_cost >= costs[next]){
					continue;
				}
				
				costs[next] = next_cost;
				if(!in_queue[next]){
					node_queue.add(next);
				}
			}
			
		}
		
		return costs;
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		final int N = sc.nextInt();
		
		ArrayList<LinkedHashSet<Integer>> adj = new ArrayList<LinkedHashSet<Integer>>();
		for(int i = 0; i < N; i++){
			adj.add(new LinkedHashSet<Integer>());
		}
		
		int[] degrees = new int[N];
		for(int i = 0; i < N - 1; i++){
			final int x = sc.nextInt() - 1;
			final int y = sc.nextInt() - 1;
		
			degrees[x]++;
			degrees[y]++;
			
			adj.get(x).add(y);
			adj.get(y).add(x);
		}
		
		Set<Integer> root = new HashSet<Integer>();
		root.add(0);
		
		Set<Integer> leafs = new HashSet<Integer>();
		for(int i = 0; i < N; i++){
			if(degrees[i] == 1){
				leafs.add(i);
			}
		}
		
		int[] root_dist = bfs(N, adj, root);
		int[] leaf_dist = bfs(N, adj, leafs);
		
		for(int i = 0; i < N; i++){
			System.out.println(Math.min(root_dist[i], leaf_dist[i]));
		}
		
	}

}
0