[C++]백준 - 5639번 문제

5639번: 이진 검색 트리 (acmicpc.net)

5639번: 이진 검색 트리

 

5693번 : 이진 검색 트리

---

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

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

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

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

 

입력

---

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

 

 

출력

---

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

 

 

 

---

 

생각해 볼 점

---

자신의 부모 노드보다 작으면 반드시 왼쪽, 크면 오른쪽에 배치되어있기 때문에,

 

전위 순회만 보아도 충분히 트리를 찾을 수 있습니다. 예제 입력 값을 보면,

 

첫 입력인 50은 루트일 것입니다.

 

30은 50보다 작으므로 왼쪽 자식입니다.

 

24는 50보다 작으므로 좌측을 탐색합니다.

 

30보다 작으므로 30의 왼쪽 자식 노드입니다.

 

입력을 받으면, Root부터 출발하여 해당 노드보다 작으면 왼쪽, 크면 오른쪽으로 탐색합니다. 만약, 끝에 도달하면 새로 노드를 추가합니다.

이렇게 트리를 완성하고 나서 후위 탐색을 실행하면 됩니다.

 

코드

---

#include <iostream>
using namespace std;

//노드 클래스
class node
{
public:
    int value;
    node *left = NULL;
    node *right = NULL;
    
    node();
    node(int v);
};

node::node(){}
node::node(int v)
{
    value = v;
}

//트리 클래스
class tree
{
private:
    node *root;
    
public:
    tree(int v);
    void insert(int v);
    void post(node *n);
    node* get_root();
};


tree::tree(int v)
{
    root = new node(v);
}

//트리에 입력
void tree::insert(int v)
{
    node *n = root;
    node *input_node = new node(v);
    
    //입력 받은 값을 Leaf에 도달할 때까지 탐색
    while(true)
    {
    	//탐색된 값보다 작으면 왼쪽
        if(v < n->value)
        {
            if(n->left == NULL) 
            {
                n->left = input_node;
                break;
            }
            else n = n->left;
        }
        //크면 오른쪽으로 탐색
        else
        {
            if(n->right == NULL) 
            {
                n->right = input_node;
                break;
            }
            else n = n->right;
        }
    }
}

//후위 순회
void tree::post(node *n)
{
    if(n->left != NULL) post(n->left);
    if(n->right !=NULL) post(n->right);
    printf("%d\n", n->value);
}

node* tree::get_root()
{
    return root;
}

int main() {
    int input;
    scanf("%d", &input);
    tree t(input);
    
    //입력된 전위 순회 값을 통해 트리를 만듬
    while(true)
    {
        if(scanf(" %d", &input) == EOF) break;
        t.insert(input);
    }
    t.post(t.get_root());
	return 0;
}

 

그 외

---