[백준 / Java] 5639번 : 이진 검색 트리

문제

이진 검색 트리는 다음과 같은 세 가지 조건을 만족하는 이진 트리이다.

• 노드의 왼쪽 서브트리에 있는 모든 노드의 키는 노드의 키보다 작다.
• 노드의 오른쪽 서브트리에 있는 모든 노드의 키는 노드의 키보다 크다.
• 왼쪽, 오른쪽 서브트리도 이진 검색 트리이다.

전위 순회 (루트-왼쪽-오른쪽)은 루트를 방문하고, 왼쪽 서브트리, 오른쪽 서브 트리를 순서대로 방문하면서 노드의 키를 출력한다. 후위 순회 (왼쪽-오른쪽-루트)는 왼쪽 서브트리, 오른쪽 서브트리, 루트 노드 순서대로 키를 출력한다. 예를 들어, 위의 이진 검색 트리의 전위 순회 결과는 50 30 24 5 28 45 98 52 60 이고, 후위 순회 결과는 5 28 24 45 30 60 52 98 50 이다.

이진 검색 트리를 전위 순회한 결과가 주어졌을 때, 이 트리를 후위 순회한 결과를 구하는 프로그램을 작성하시오.

입력

트리를 전위 순회한 결과가 주어진다. 노드에 들어있는 키의 값은 106보다 작은 양의 정수이다. 모든 값은 한 줄에 하나씩 주어지며, 노드의 수는 10,000개 이하이다. 같은 키를 가지는 노드는 없다.

출력

입력으로 주어진 이진 검색 트리를 후위 순회한 결과를 한 줄에 하나씩 출력한다.

예제 입력 1 복사

50
30
24
5
28
45
98
52
60

예제 출력 1 복사

5
28
24
45
30
60
52
98
50

https://www.acmicpc.net/problem/5639

5639번: 이진 검색 트리

---

문제 풀이

 해당 문제는 제목 그대로 이진 검색 트리를 구현하는 문제이다. 이진 검색 트리는 각 Node 기준으로 낮은 수의 경우 왼쪽 노드로, 높은 수의 경우 오른쪽 노드로 설정하는 트리 구조이다. 문제에서 전위 순회한 결과를 주어주므로 단순히 트리에 삽입을 하면 원래 노드가 구현된다.

 구현된 노드를 통해 후위 순회한 결과를 출력하면 된다.

 

노드 구현 코드

public static class Node{
		Node left;
		Node right;
		int num;
		
		public Node(int num) {
			left = null;
			right = null;
			this.num = num;
		}
		
		public void addNode(Node newNode) {
			if(this.num > newNode.num) {
				if(this.left == null) {
					this.left = newNode;
				}else {
					this.left.addNode(newNode);
				}
			}else {
				if(this.right == null) {
					this.right = newNode;
				}else {
					this.right.addNode(newNode);
				}
			}
		}
	}

 

후위  순회 출력 메서드 코드

public static void preoder(Node N) {
		if(N == null)
			return;
		
		preoder(N.left);
		preoder(N.right);
		System.out.println(N.num);
}

---

풀이 코드

import java.util.*;
import java.io.*;
public class Main {
	public static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	public static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	

	public static class Node{
		Node left;
		Node right;
		int num;
		
		public Node(int num) {
			left = null;
			right = null;
			this.num = num;
		}
		
		public void addNode(Node newNode) {
			if(this.num > newNode.num) {
				if(this.left == null) {
					this.left = newNode;
				}else {
					this.left.addNode(newNode);
				}
			}else {
				if(this.right == null) {
					this.right = newNode;
				}else {
					this.right.addNode(newNode);
				}
			}
		}
	}
	
	public static void main(String[] args) throws IOException{
		Node root = new Node(Integer.parseInt(br.readLine()));
		
		while(true) {
			String temp = br.readLine();
			
			if(temp == null || temp.equals(""))break;
			root.addNode(new Node(Integer.parseInt(temp)));
			
		}
		preoder(root);
	}
	
	public static void preoder(Node N) {
		if(N == null)
			return;
		
		preoder(N.left);
		preoder(N.right);
		System.out.println(N.num);
	}
	
}