Problem:
MathJax.Hub.Config({ tex2jax: { inlineMath: [["$","$"], ["\\(","\\)"]], processEscapes: true }});

Binary Search Tree I

Search trees are data structures that support dynamic set operations including insert, search, delete and so on. Thus a search tree can be used both as a dictionary and as a priority queue.

Binary search tree is one of fundamental search trees. The keys in a binary search tree are always stored in such a way as to satisfy the following binary search tree property:

Let $x$ be a node in a binary search tree. If $y$ is a node in the left subtree of $x$, then $y.key \leq x.key$. If $y$ is a node in the right subtree of $x$, then $x.key \leq y.key$.

The following figure shows an example of the binary search tree.

For example, keys of nodes which belong to the left sub-tree of the node containing 80 are less than or equal to 80, and keys of nodes which belong to the right sub-tree are more than or equal to 80. The binary search tree property allows us to print out all the keys in the tree in sorted order by an inorder tree walk.

A binary search tree should be implemented in such a way that the binary search tree property continues to hold after modifications by insertions and deletions. A binary search tree can be represented by a linked data structure in which each node is an object. In addition to a key field and satellite data, each node contains fields left, right, and p that point to the nodes corresponding to its left child, its right child, and its parent, respectively.

To insert a new value $v$ into a binary search tree $T$, we can use the procedure insert as shown in the following pseudo code. The insert procedure is passed a node $z$ for which $z.key = v$, $z.left = NIL$, and $z.right = NIL$. The procedure modifies $T$ and some of the fields of $z$ in such a way that $z$ is inserted into an appropriate position in the tree.

1 insert(T, z)
2 y = NIL // parent of x
3 x = 'the root of T'
4 while x ≠ NIL
5 y = x // set the parent
6 if z.key < x.key
7 x = x.left // move to the left child
8 else
9 x = x.right // move to the right child
10 z.p = y
11
12 if y == NIL // T is empty
13 'the root of T' = z
14 else if z.key < y.key
15 y.left = z // z is the left child of y
16 else
17 y.right = z // z is the right child of y

Write a program which performs the following operations to a binary search tree $T$.

insert $k$: Insert a node containing $k$ as key into $T$.

print: Print the keys of the binary search tree by inorder tree walk and preorder tree walk respectively.

You should use the above pseudo code to implement the insert operation. $T$ is empty at the initial state.

Input

In the first line, the number of operations $m$ is given. In the following $m$ lines, operations represented by insert $k$ or print are given.

Output

For each print operation, print a list of keys obtained by inorder tree walk and preorder tree walk in a line respectively. Put a space character before each key.

Constraints

The number of operations $\leq 500,000$

The number of print operations $\leq 10$.

$-2,000,000,000 \leq key \leq 2,000,000,000$

The height of the binary tree does not exceed 100 if you employ the above pseudo code.

The keys in the binary search tree are all different.

Sample Input 1

8
insert 30
insert 88
insert 12
insert 1
insert 20
insert 17
insert 25
print

Sample Output 1

1 12 17 20 25 30 88
30 12 1 20 17 25 88

Reference

Introduction to Algorithms, Thomas H. Cormen, Charles E. Leiserson, Ronald L. Rivest, and Clifford Stein. The MIT Press.

Solution in C (judge verdict: Accepted)

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct node {
    int value;
    struct node* parent;
    struct node* left;
    struct node* right;
} Node;

typedef Node* NodePointer;

NodePointer serialize(int val) {
    NodePointer newNode = malloc(sizeof(Node));
    newNode->value = val;
    newNode->parent = NULL;
    newNode->left = NULL;
    newNode->right = NULL;
    return newNode;
}

NodePointer insert(NodePointer node, int val) {
    NodePointer currentNode = node;
    int flag = 0;
    int dir = 0;
    while(!flag) {
        if (currentNode->value > val) {
            if (currentNode->left == NULL) flag = 1;
            else currentNode = currentNode->left;
            dir = 0;
        } else if (currentNode->value < val) {
            if (currentNode->right == NULL) flag = 1;
            else currentNode = currentNode->right;
            dir = 1;
        } else {
            return node;
        }
    }

    if (dir) {
        currentNode->right = serialize(val);
        currentNode->right->parent = currentNode;
    } else {
        currentNode->left = serialize(val);
        currentNode->left->parent = currentNode;
    }

    return node;
}

int search(NodePointer node, int val) {
    NodePointer currentNode = node;
    while(currentNode != NULL) {
        if (currentNode->value == val) {
            return 1;
        } else if (currentNode->value > val) {
            currentNode = currentNode->left;
        } else {
            currentNode = currentNode->right;
        }
    }

    return 0;
}

NodePointer min(NodePointer node) {
    NodePointer currentNode = node;
    int flag = 0;
    while (!flag) {
        if (currentNode->left == NULL) flag = 1;
        else currentNode = currentNode->left;
    }

    return currentNode;
}

NodePointer delete(NodePointer node, int val) {
    NodePointer currentNode = node;
    while(currentNode->value != val) {
        if (currentNode->value > val) {
            if (currentNode->left == NULL) return node;
            currentNode = currentNode->left;
        } if (currentNode->value < val) {
            if (currentNode->right == NULL) return node;
            currentNode = currentNode->right;
        }
    }

    // ノードに子がいない
    if (currentNode->right == NULL && currentNode->left == NULL) {
        if (currentNode->parent->left == currentNode) currentNode->parent->left = NULL;
        else currentNode->parent->right = NULL;
    // ノードの左にだけ子がいる
    } else if (currentNode->right == NULL) {
        if (currentNode->parent->left == currentNode) {
            currentNode->parent->left = currentNode->left;
            currentNode->left->parent = currentNode->parent;
        } else {
            currentNode->parent->right = currentNode->left;
            currentNode->left->parent = currentNode->parent;
        }
    // ノードの右にだけ子がいる
    } else if (currentNode->left == NULL) {
        if (currentNode->parent->left == currentNode) {
            currentNode->parent->left = currentNode->right;
            currentNode->right->parent = currentNode->parent;
        } else {
            currentNode->parent->right = currentNode->right;
            currentNode->right->parent = currentNode->parent;
        } 
    // どちらにも子がいる
    } else {
        NodePointer minPtr = min(currentNode->right);
        if (minPtr->parent->left == minPtr) minPtr->parent->left = NULL;
        else minPtr->parent->right = NULL;
        if (currentNode->parent->left == currentNode) {
            currentNode->parent->left = minPtr;
            minPtr->parent = currentNode->parent;
        } else {
            currentNode->parent->right = minPtr;
            minPtr->parent = currentNode->parent;
        }
        minPtr->left = currentNode->left;
        minPtr->right = currentNode->right;
    }
    free(currentNode); 

    return node;
}

void inorderPrint(NodePointer node) {
    if (node == NULL) return;
    inorderPrint(node->left);
    printf(" %d", node->value);
    inorderPrint(node->right);
}

void preorderPrint(NodePointer node) {
    if (node == NULL) return;
    printf(" %d", node->value);
    preorderPrint(node->left);
    preorderPrint(node->right);
}

void nodePrint(NodePointer node) {
    inorderPrint(node);
    printf("\n");
    preorderPrint(node);
    printf("\n");
}

int main() {
    NodePointer node = NULL;
    int n, m;
    char cmd[16];
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%s", cmd);
        if (!strcmp(cmd, "insert")) {
            scanf("%d", &m);
            if (node == NULL) node = serialize(m);
            else insert(node, m);
        } else if (!strcmp(cmd, "print")) {
            nodePrint(node);
        } else if (!strcmp(cmd, "find")) {
            scanf("%d", &m);
            if (search(node, m)) {
                printf("yes\n");
            } else {
                printf("no\n");
            }
        } else if (!strcmp(cmd, "delete")) {
            scanf("%d", &m);
            delete(node, m);
        }
    }
    return 0;
}