본문 바로가기

백준/그래프

이진검색트리 - 5639번 (java)

728x90

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

 

5639번: 이진 검색 트리

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

www.acmicpc.net

문제

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

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

전위 순회 (루트-왼쪽-오른쪽)은 루트를 방문하고, 왼쪽 서브트리, 오른쪽 서브 트리를 순서대로 방문하면서 노드의 키를 출력한다. 후위 순회 (왼쪽-오른쪽-루트)는 왼쪽 서브트리, 오른쪽 서브트리, 루트 노드 순서대로 키를 출력한다. 예를 들어, 위의 이진 검색 트리의 전위 순회 결과는 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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Scanner;

public class Main {
	static Node root;
	static StringBuilder sb = new StringBuilder();
	static class Node{
		
		int num;
		Node left;
		Node right;
		
		Node(int num){
			this.num = num;
		}
		
	}
	
	
	public static void putNode(Node startNode, Node inputNode) {
		if(startNode.num > inputNode.num) {
			if(startNode.left!=null) {
				putNode(startNode.left, inputNode);
			}
			else {
				startNode.left = inputNode;
			}
		}
		else {
			if(startNode.right != null) {
				putNode(startNode.right, inputNode);
			}
			else {
				startNode.right = inputNode;
			}
		}
	}
	
	public static void postOrder(Node node) {
		
		if(node.left!=null) {postOrder(node.left);}
		if(node.right!=null) {postOrder(node.right);}
		sb.append(node.num+"\n");
		
	}
	
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		
		Scanner sc = new Scanner(System.in);
		root = new Node( sc.nextInt());

		while(sc.hasNext()) {
			putNode(root, new Node(sc.nextInt()));
		}
		postOrder(root);
		System.out.println(sb);
	}
	

}

- 전위 순회 -> 후위순회 를 위해서는 이진검색트리를 링크드리스트 방식으로 구현을 해야한다. 

'백준 > 그래프' 카테고리의 다른 글

최단경로 -1753번  (0) 2021.08.25
트리의 부모찾기  (0) 2021.04.26
키 순서 / 플로이드 와샬  (0) 2021.04.24