Delete node in BST

最新推荐文章于 2019-04-20 04:23:21 发布

ActiveCoder

最新推荐文章于 2019-04-20 04:23:21 发布

阅读量367

点赞数

分类专栏： LintCode 题解

本文链接：https://blog.csdn.net/github_34333284/article/details/52134087

版权

LintCode 题解专栏收录该内容

6 篇文章 0 订阅

订阅专栏

Question: delete the given value in BST.

This question is pretty tricky, to delete a node in BST, there are several cases to consider.

1: The node is one of the leaves, 2: the node only has right children, 3: the node only has left children, 4: the node has left and right children.

code:

TreeNode* deleteTreeNode(TreeNode* root, int val) {
  if(!root) return NULL;
  else {
    if(val < root->val) root->left = deleteTreeNode(root->left, val);
    else if(val > root->val) root->right = deleteTreeNode(root->right, val);
    else {
      // if there is no left child, no right child
      if(root->left == NULL && root->right == NULL) {
        delete root;
        root = NULL;
        return root;
      }
      else if(root->left == NULL) { // if there is no left subtree
        TreeNode* tmp = root;
        root = root->right;
        delete tmp;
        return root;
      } else if(root->right == NULL) { // if there is no right subtree
        TreeNode* tmp = root;
        root = root->right;
        delete tmp;
        return root;
      } else { // two children
        struct TreeNode* tmp = findMin(root->right);
        root->data = tmp->data;
        root->right = deleteTreeNode(root->right, tmp->data);
        return root;
      }
    }
  }
}