[백준] 11725: 트리의 부모 찾기

SuKong·2020년 8월 24일

그래프 백준 알고리즘 트리

BOJ의 알고리즘 문제를 풀어보아요

목록 보기

38/49

'11725- 트리의 부모 찾기' 문제로 이동!

👉문제

루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.

👉입력

첫째 줄에 노드의 개수 N (2 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N-1개의 줄에 트리 상에서 연결된 두 정점이 주어진다.

예시 -
7
1 6
6 3
3 5
4 1
2 4
4 7

👉출력

첫째 줄부터 N-1개의 줄에 각 노드의 부모 노드 번호를 2번 노드부터 순서대로 출력한다.

예시 -
4
6
1
3
1
4

✍내 풀이

import java.util.*;

public class Main {	
	static ArrayList<Integer>[] tree;
	static boolean[] isVisited;
	static int[] returnArr;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int nodeNum = sc.nextInt();
		tree = new ArrayList[nodeNum + 1];
		isVisited = new boolean[nodeNum + 1];
		returnArr = new int[nodeNum+1];
		
		for( int i = 1 ; i <= nodeNum ; i++ ) {
			tree[i] = new ArrayList<Integer> ();
			isVisited[i] = false;
		}
		
		for( int i = 0 ; i < nodeNum-1 ; i++ ) {
			int a = sc.nextInt();
			int b = sc.nextInt();
			tree[a].add(b);
			tree[b].add(a);
		}
		DFS(1);

		for( int i = 2 ; i <= nodeNum ; i++ ) {
			System.out.println(returnArr[i]);
		}
	}
	
	static void DFS(int node) {
		isVisited[node] = true;
		for( int i = 0 ; i < tree[node].size() ; i++ ) {
			int adjNode = tree[node].get(i);
			if( isVisited[adjNode] == false ) {
				returnArr[adjNode] = node;
				DFS(adjNode);
			}
		}
	}
}