18-YIM2UL2ET

[YIM2UL2ET]

🔗 문제 링크

이진 검색 트리 - BOJ 5639

✔️ 소요된 시간

2h

✨ 수도 코드

1. 문제 설명

• 노드의 왼쪽 서브트리에 있는 모든 노드의 키는 노드의 키보다 작다.
• 노드의 오른쪽 서브트리에 있는 모든 노드의 키는 노드의 키보다 크다.
• 왼쪽, 오른쪽 서브트리도 위 조건을 만족하는 트리이다

이러한 트리를 전위순회한 결과를 입력받고, 후위순위를 하여 출력하면 된다.

2. 문제 분석

입력 받기

전위 순회 방식은 루트노드 -> 좌측노드 -> 우측노드 이다.
그렇다면 입력도 루트노드부터 입력을 받으면 될 것이다.

노드의 값을 입력 받을 때 노드들은 전부 다음 조건을 만족시켜야 한다.

• 오른쪽 서브트리는 자신의 값보다 모두 크다.
• 왼쪽 서브트리는 자신의 값보다 모두 작다.

따라서 루트노드에 값이 존재한다면 왼쪽 자식 노드와, 오른쪽 자식 노드를 입력받아야 하는데, 이 조건을 이용하여 입력받으면 될 것이다.

Input 함수 (노드의 주소, 값):
    if 매개변수로 입력받은 노드에 값이 없으면: 
        노드에 값을 입력
    else if 매개변수 노드의 값 > 매개변수 값 :
        Input(좌측 노드의 주소, 값)
    else
        Input(우측 노드의 주소, 값)

출력하기

입력을 다 받았다면 트리가 완성되었을 것이므로 루트 노드의 주소를 이용하여 후위순회를 하면 된다.
후위 순회는 좌측노드 -> 우측 노드 -> 루트 노드 이다.
이는 간단하게 구현이 가능하다.

Output 함수 (노드의 주소):
    if 매개변수로 입력받은 노드에 왼쪽 자식 노드가 있다면:
        Output(좌측 노드의 주소)
    if 매개변수로 입력받은 노드에 오른쪽 자식 노드가 있다면:
        Output(우측 노드의 주소)
    매개변수로 입력받은 노드의 값 출력

3. 최종 코드

#include <iostream>
using namespace std;

struct node {
    int value = -1;
    node* right = NULL;
    node* left = NULL;
};

void input(node &nd, int n) {
    if (nd.value == -1) nd.value = n;
    else if (nd.value > n) {
        if (nd.left == NULL) nd.left = new node;
        input(*nd.left, n); 
    }
    else {
        if (nd.right == NULL) nd.right = new node;
        input(*nd.right, n); 
    }
}

void output(node &nd) {
    if (nd.left != NULL) output(*nd.left);
    if (nd.right != NULL) output(*nd.right);
    cout << nd.value << '\n';
}

int main(void) {
    int n;
    node root;
    while (cin >> n) input(root, n);
    output(root);
    return 0;
}

📚 새롭게 알게된 내용

트리에서 입력 받는 테크닉: 순회 방식을 잘 보고 입력 받는 방식을 찾자.
전위순회는 루트노드에서 입력받으면 굿