배열을 이용한 이진 탐색 트리(Binary Search Tree) 소스 코드

insert(number X, node N)

if X가 노드 N에 있는 숫자보다 작다면

if N의 왼쪽 자식이 없다면

X를 포함하는 새 노드를 만든 뒤, N의 왼쪽 자식으로 만든다.

else

insert(X, N의 왼쪽 자식)

else X가 노드 N에 있는 숫자보다 크다면

if N의 오른쪽 자식이 없다면

X를 포함하는 새 노드를 만든 뒤, N의 오른쪽 자식으로 만들기

else

insert(X, N의 오른쪽 자식)

http://www.crocus.co.kr/447 에 이전에 수록한 코드를 포인터가 아닌 배열로 구현한 코드이다.

소스 코드 : 

#include <iostream>
#include <cstdio>
 
using namespace std;
 
typedef struct _TREE_
{
    int idx;
    int left;
    int right;
    int parent;
}TREE;
 
TREE tree[200002];
 
int main()
{
    int n, val, root;
    scanf("%d", &n);
 
    int tmp;
    int cnt = 0;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &val);
        if (i == 1)
        {
            tree[val].idx = i;
            root = val;
            tmp = root;
        }
        else
        {
            tree[val].idx = i;
            root = tmp; // 최고 루트 노드
            while(1)
            {
                cnt++;
                if(val < root)
                {
                    if (tree[root].left == 0)
                    {
                        tree[val].parent = root;
                        tree[root].left = val;
                        break;
                    }
                    else
                        root = tree[root].left;
                    
                }
 
                else if (val > root)
                {
                    if (tree[root].right == 0)
                    {
                        tree[val].parent = root;
                        tree[root].right = val;
                        break;
                    }
                    else
                        root = tree[root].right;
                }
            }
        }
 
        printf("%d\n", cnt);
    }
 
    //for (int i = 0; i <= n; i++)
    //    cout << "val :: " << i << " idx ::" << tree[i].idx << " parent :: " << tree[i].parent << " left :: " << tree[i].left << " right :: " << tree[i].right << endl;
    
    return 0;
}
 
//                                                       This source code Copyright belongs to Crocus
//                                                        If you want to see more? click here >>