[HackerRank] Binary Search Tree : Insertion

 

7주차 자료구조 복습

풀이

먼저 트리의 루트가 있는지 확인해 준다. 이때 루트가 존재하지 않는다면, 루트에 동적할당을 한 뒤, 데이터를 넣고 left와 right은 각각 NULL로 초기화 한다. 

루트가 있는 경우엔 데이터 값의 크기에 따라 구분해야 한다. 삽입할 값이 루트의 데이터보다 작다면 left에 넣어야 하므로 insert 함수에 root->left를 불러 데이터를 넣어준다. 반대의 경우 right을 불러 데이터를 넣어주면 된다.

마지막엔 root를 반환한다.

    Node * insert(Node * root, int data) {
        if(root == NULL){
            root->data = data;
            root->left=NULL;
            root->right = NULL;
        }
        else if(data < root->data)
            root->left = insert(root->left, data);
        else {
            root->right = insert(root->right, data);
        }
        return root;
    }

결과