算法-二叉查找树-删除节点

题目:给定一棵具有不同节点值的二叉查找树,删除树中与给定值相同的节点。如果树中没有相同值的节点,就不做任何处理。你应该保证处理之后的树仍是二叉查找树。

样例:

给出如下二叉查找树:

          5

       /    \

    3          6

 /    \

2       4

删除节点3之后,你可以返回:

          5

       /    \

    2          6

      \

         4

或者:

          5

       /    \

    4          6

 /   

2

代码如下:

当要删除的结点存在左右子树,首先要找到该节点的后继节点(左子树中最大的/右子树中最小的)


/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /*
     * @param root: The root of the binary search tree.
     * @param value: Remove the node with given value.
     * @return: The root of the binary search tree after removal.
     */
    public TreeNode removeNode(TreeNode root, int value) {
        // write your code here
    if(root == null)
    return root;
    if(value > root.val)
    root.right = removeNode(root.right,value);
    else if(value < root.val)
    root.left = removeNode(root.left,value);
    else if(root.left != null && root.right != null) //上述情况
    {
        root.val = findMin(root.right).val;
        root.right = removeNode(root.right,root.val);
    }
    else
    root = (root.left != null) ? root.left :root.right;//  三种情况:左子树非空,右子树空;左子树空,右子树非空;左子树空,右子树空。
    return root;
    }
    public TreeNode findMin(TreeNode t){
        if(t == null)
        return t;
        else if(t.left == null)
        return t;
        return findMin(t.left);
    }
}


    原文作者:查找算法
    原文地址: https://blog.csdn.net/Ailubby/article/details/79212092
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞