백준_5693_이진 검색 트리[C++]

[문제 바로가기]

문제

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

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

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

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

풀이

1. BST를 구현한다.
2. 입력은 전휘 순회의 결과로 들어온다. 차례대로 BST에 넣어준다.
3. 후휘 순회 결과를 출력한다.

코드

#include<iostream>
using namespace std;
class Node{
public:
    int val;
    Node *left;
    Node *right;
    Node * parent;
    Node(int val){
        this->val =val;
        left = right = parent =NULL;
    }
    void setLeft(Node *node){
        this->left = node;
        return;
    }
    Node* getLeft(){
        return this->left;
    }
    void setRight(Node * node){
        this->right=node;
        return;
    }
    Node* getRight(){
        return this->right;
    }
    void setParent(Node *node){
        this->parent=node;
        return;
    }
    Node* getParent(){
        return this->parent;
    }
};
class BST{
public:
    Node *root;
    BST(){
        root=NULL;
    }
    void insertNode(Node *node){
        if(root==NULL){
            root = node;
            return;
        }
        Node* cur_node = root;
        Node* par_node;
        while(cur_node!=NULL){
            if(cur_node->val<node->val){
                par_node = cur_node;
                cur_node = cur_node->getRight();
            }
            else{
                par_node = cur_node;
                cur_node = cur_node->getLeft();
            }
        }
        if(par_node->val<node->val)
            par_node->setRight(node);
        else par_node->setLeft(node);
    }
    void print_Post(Node * node){
        if(node==NULL)
            return;
        print_Post(node->getLeft());
        print_Post(node->getRight());
        cout<<node->val<<"\n";

    }
};
int main(){
    int num;
    BST bst;
    while(cin>>num){
        bst.insertNode(new Node(num));
    }
    bst.print_Post(bst.root);
}