Red-Black 트리

Red-Black 트리는 self-balancing binary search tree이다.

레드-블랙 트리를 포함한 이진 탐색 트리는, 모든 노드에 대해 '자신이 가진 자료는 자신보다 오른쪽에 위치한 부분트리가 가지고 있는 모든 자료보다 작거나 같고, 자신보다 왼쪽에 위치한 부분트리가 가지고 있는 모든 자료보다 크거나 같다' 라는 조건을 만족한다. 이런 특성 때문에 특정 값을 빠르게 찾아 낼 수 있으며, 각 구성원소(elements)간의 효율적인 in-order traversal이 가능하다.

정의

노드는 레드 혹은 블랙 중 하나이다.
Root Property : 루트노드는 블랙이다.
External Property : 모든 리프노드들은 검정(Black)이다.
Internal Property : 빨강(Red)노드의 자식노드는 언제나 검정(Black)이다.
- 즉, 레드 노드는 연달아 나타날 수 없으며, 블랙 노드만이 레드노드의 부모가 될 수 있다.
Depth Property : 모든 단말노드에서 Black Height는 같다.
- 어떤 노드로부터 시작되어 리프노드에 도달하는 모든 경로는 리프노드를 제외하면 모두 같은 수의 블랙노드를 가지고 있다.
- 노드의 Height h일때, black-height >= h/2.

Red-Black 트리는 Binary Search Tree이므로 BST의 특징을 모두 갖는다.

루트 노드부터 가장 먼 경로까지의 거리가, 가장 가까운 경로까지의 거리의 두 배 보다 항상 작다. 이러한 상태를 balanced상태라고 한다. 따라서, 삽입, 삭제, 검색시 최악의 경우 시간복잡도가 트리의 높이(또는 깊이)에 따라 결정되기 때문에 보통의 이진 탐색 트리에 비해 효율적이라고 할 수 있다.

동작

Red-Black Tree에서는 1. Recoloring 2. Rotation 두 가지 방법을 이용해서 균형을 맞춘다. 우선 recoloring을 한 후, recoloring이 불가능해지면, rotation을 한다.

삽입(insertion)

x를 새로 삽입된 노드라고 가정할 것이다.

Binary Search Tree와 같이 노드를 삽입한다. 그리고 새로 삽입되는 노드 x를 RED로 색을 정한다.

만약 x가 root노드이면, x의 색을 BLACK으로 변경한다. (black-height 는 1씩 증가)

void insert_case1(Node * n)
{
    if (n->parent == NULL)
        n->color = BLACK;
    else
        insert_case2(n);
}

x의 부모 색상이 BLACK이 아니거나 x가 root 노드가 아닌 경우

redBlackCase2

x 의 삼촌노드가 RED인 경우( 속성4 : Grand parent는 반드시 검정색이어야한다. )

void insert_case2(Node * n)
{
    if (n->parent->color == BLACK)
        return; /* Tree is still valid */
    else
        insert_case3(n);
}
void insert_case3(Node * n)     // 이미 부모는 R이 판명된 상태
{
    if (uncle(n) != NULL && uncle(n)->color == RED)
    {
        n->parent->color = BLACK;
        uncle(n)->color = BLACK;
        grandparent(n)->color = RED;

        insert_case1(grandparent(n));
    }
    else
        insert_case4(n);
}

부모와 삼촌의 색을 BLACK으로 변경
조상(Grand Parent)색은 RED로 변경
x = x의 Grand Parent로 변경하고 2와 3을 반복해서 진행한다.

x의 삼촌노드가 BLACK인 경우(AVL Tree와 유사)

Left Left Case(case 5)
Left Right Case(case4)
Right Right Case

Right Left Case

// 삽입된 노드와 부모, 부모의 부모가 일련의 왼쪽 라인을 타도록...
void insert_case4(node n)      
{
    if (n == n->parent->right && n->parent == grandparent(n)->left)
    {
        rotate_left(n->parent);
        n = n->left;
    }
    else if (n == n->parent->left && n->parent == grandparent(n)->right)
    {
        rotate_right(n->parent);
        n = n->right;
    }
    insert_case5(n);
}

// 노드와 부모와 부모의 부모, 모두 같은 방향의 자식 관계라면... 색 반전하면서 회전
void insert_case5(node n)
{
    n->parent->color = BLACK;
    grandparent(n)->color = RED;

    if (n == n->parent->left && n->parent == grandparent(n)->left)
    {
        rotate_right(grandparent(n));
    }
    else
    {
        /* Here, n == n->parent->right && n->parent == grandparent(n)->right */
        rotate_left(grandparent(n));
    }
}

예제

코드

#include "redblack.h"
#define RED 1
#define BLACK 0

typedef struct node{
    int color;
    int key;
    struct node * left, * right, * parent;
}Node;

Node * new_node(int key){
    Node * new = malloc(sizeof(Node));
    new->left = new->right = new->parent = NULL;
    new->key=key;
    new->color=RED;
    return new;
}

void rotate_left(Node * node,Node ** root)
{
    printf("left rotation!!\n");
    Node * r = node->right;
    Node * p = node->parent;

    if(r->left!=NULL){
        r->left->parent = node;
    }
    node->right=r->left;
    node->parent=r;
    r->left=node;
    r->parent=p;


    if(p==NULL){
        *root = r;
    }else if(node == p->left){
        p->left = r;
    }else{
        p->right = r;
    }
}


void rotate_right(Node * node,Node ** root)
{
    printf("right rotation!!\n");
    Node * l = node->left;
    Node * p = node->parent;

    if(l->right!=NULL){
        l->right->parent = node;
    }
    node->left=l->right;
    node->parent=l;
    l->right=node;
    l->parent=p;

    if(p==NULL){
        *root = l;
    }else if(node == p->left){
        p->left = l;
    }else{
        p->right = l;
    }
}

Node * grand_node(Node *n){
    if ((n != NULL) && (n->parent != NULL))
        return n->parent->parent;
    else
        return NULL;
}
Node *uncle_node(Node *n)
{
    struct node *g = grand_node(n);
    if (g == NULL)
        return NULL; 
    if (n->parent == g->left)
        return g->right;
    else
        return g->left;
}
void insert_fix(Node ** root, Node * node){

    if(node->parent==NULL){
        node->color=BLACK;
        return;
    }else{
        if(node->parent->color==RED){
            Node * grand = node->parent->parent;
            Node * uncle;
            if(grand->left==node->parent)uncle=grand->right;
            else uncle=grand->left;

            if(uncle==NULL){
                grand->color=RED;
                node->parent->color=BLACK;
                if(node->parent==grand->left){
                    if(node==node->parent->left){
                        rotate_right(grand, root);
                        grand->left=NULL;
                    }else{
                        rotate_left(node->parent, root);
                        rotate_right(grand, root);
                    }

                }else{
                    if(node == node->parent->right){
                        rotate_left(grand, root);
                        grand->right=NULL;
                    }else{
                        rotate_right(node->parent, root);
                        rotate_left(grand, root);
                    }

                }
            }
            else if(uncle->color==RED){
                node->parent->color=BLACK;
                uncle->color=BLACK;

                grand->color=RED;
                node=grand;
            }else{
                if(node->parent==grand->left){
                    if(node==node->parent->right){
                        node=node->parent;
                        rotate_left(node, root);
                    }
                    node->parent->color=BLACK;
                    grand->color=RED;

                    rotate_right(grand, root);
                }else{
                    if(node==node->parent->left){
                        node=node->parent;
                        rotate_right(node, root);
                    }
                    node->parent->color=BLACK;
                    node->parent->parent->color=RED;

                    rotate_left(node->parent->parent, root);
                }
            }
        }
    }

    (*root)->color=BLACK;
}

void insert_node(Node ** root, int key){

    Node * new = new_node(key);

    Node * tmp = *root;
    Node * p=NULL;

    while(tmp!=NULL){
        p = tmp;
        if(key>tmp->key)tmp=tmp->right;
        else tmp=tmp->left;
    }
    new->parent = p;
    if(p==NULL)*root = new;
    else if(key>p->key){
        p->right = new;
    }
    else{
        p->left = new;
    }

    insert_fix(root, new);

}
void print_tree(Node *root, int space){
    if (root == NULL)
        return;
    space += 10; // level이 클수록 오른쪽에 나타나도록 space증가

    print_tree(root->right, space);
    printf("\n");
    for(int i=10; i<space; i++)printf(" ");
    printf("%d[%d]\n",root->key,root->color);
    print_tree(root->left, space);
}
int main(){
    Node * root = NULL;
    insert_node(&root, 2);
    insert_node(&root, 1);
    insert_node(&root, 4);
    insert_node(&root, 5);
    insert_node(&root, 9);
    insert_node(&root, 3);
    insert_node(&root, 6);
    insert_node(&root, 7);
    print_tree(root, 0);
}

삭제

삽입과 마찬가지로 1. Recoloring 2. Rotation으로 Red-Black의 속성을 유지해야한다. 삽입에서는 삼촌노드의 색을 확인했는데, 삭제에서는 형제노드의 색을 확인해야한다. 삭제는 더 복잡하다. 검은색 노드를 삭제했을 때, 검정색 하위 노드로 대체된다면, 자식노드는 double black이다. 삭제에서 주요한 작업은 double black을 단일 black으로 변환하는 것이다.

삭제할 노드를 v라 가정

BST와 같이 노드를 삭제한다. BST에서 삭제는 자식이 1개이하일때 노드를 삭제하므로 노드가 리프노드이거나 자식이 1개인 경우만 처리하면된다. v를 삭제하고 v를 자식노드로 대체한다

단순한 경우 : 형제노드가 없는 경우(u 또는 v가 RED이면 우리는 대체한 자식노드의 색을 Black으로 바꿔준다.)

 /*
  * 선제조건: n이 최대 하나의 non-null 자식을 갖고 있음.
  */
void delete_one_child(node *n){
 node *child = is_leaf(n->right) ? n->left : n->right;

 replace_node(n, child);
 if (n->color == BLACK) {
  if (child->color == RED)
   child->color = BLACK;
  else
   delete_case1(child);
 }
 free(n);
}

형제노드가 있는 경우

Case1 : Node가 새로운 루트가 되는 경우
- 새로운 루트는 검은색이므로 모든 특성이 보존되므로 통과
```
void delete_case1(struct node *n)
{
 if (n->parent != NULL)
  delete_case2(n);
}
```

Case2 : 형제 노드 색이 Red인 경우, 형제 노드를 Rotate한 후 부모노드와 형제노드 Recoloring한다.

Left Case(Right Case와 반대)

Right Case

void delete_case2(struct node *n)
{
 struct node *s = sibling(n);

 if (s->color == RED) {
  n->parent->color = RED;
  s->color = BLACK;
  if (n == n->parent->left)
   rotate_left(n->parent);
  else
   rotate_right(n->parent);
 }
 delete_case3(n);
}

Case3 : 형제 노드 색이 Black이고 형제노드의 자식노드가 모두 Black인 경우 : Recoloring 수행

void delete_case3(struct node *n)
{
 struct node *s = sibling(n);

 if ((n->parent->color == BLACK) &&
     (s->color == BLACK) &&
     (s->left->color == BLACK) &&
     (s->right->color == BLACK)) {
  s->color = RED;
  delete_case1(n->parent);
 } else
  delete_case4(n);
}

Case4 : 형제노드와 형제노드의 자식노드는 검은색이지만, 부모노드는 빨간색인경우

부모노드와 형제노드의 색을 바꿔준다.

void delete_case4(struct node *n)
{
 struct node *s = sibling(n);

 if ((n->parent->color == RED) &&
     (s->color == BLACK) &&
     (s->left->color == BLACK) &&
     (s->right->color == BLACK)) {
  s->color = RED;
  n->parent->color = BLACK;
 } else
  delete_case5(n);
}

Case5 : 형제노드의 색이 Black이고, 형제 노드의 자식노드 중 적어도 한개의 노드가 Red이면 rotate한다.
- Left Left Case
- Left Right Case
- Right Right Case
- Right Left Case

시간복잡도

Red-Black 트리의 search, insert, delete 연산은 O(logn)의 시간복잡도가 소요된다. 동일한 노드의 개수일 때, depth를 최소화해 시간 복잡도를 줄이는 것이 핵심인 알고리즘이다. 이때 depth가 최소가 되는 경우는 tree가 complete binary tree인 경우이다.

용도

대표적으로 연관 배열(associative array)를 구현하는데 사용한다.

키 하나와 값 하나가 연관되어 있으며 키를 통해 연관되는 값을 얻을 수 있다.
연상 배열, 결합형 배열, 맵(map), 사전(dictionary)으로 부르기도한다.

AVL Tree 와 차이점

AVL 트리는 Red Black Trees에 비해 균형이 잡혀 있지만 삽입 및 삭제 중에 더 많은 회전이 발생할 수 있다. 따라서 응용 프로그램에 삽입 및 삭제가 자주 발생하면 Red Black Tree를 사용하는 것이 좋다. 삽입 및 삭제 횟수가 적고 검색 빈도가 높으면 AVL 트리가 Red Black Tree보다 효율성이 높다.

Previous다익스트라 알고리즘(Dijkstra's Algorithm)NextA* 알고리즘

Last updated 3 years ago