LeetCode450.删除二叉搜索树中的节点 Java

题目描述

给定一个二叉搜索树的根节点root和一个值key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。

思路

如题目描述中所说，先找到节点，再删除节点。
由于要保持二叉搜索树的特点，即左节点的值均小于本节点的值，右节点的值均大于本节点的值。
因此，在删除对应节点后，要注意将该节点的左子树接到右子树的最左侧。
此外，如果删除的是根节点，则需要将根节点的子节点作为新的根节点返回。

代码

首先，在main函数中进行讨论，如果要删除的是根节点root，则直接调用insert函数，返回新的根节点；如果要删除的不是根节点，则通过深搜dfs函数进行遍历，若遇到子节点值为key，则调用insert，返回新的子节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if( root==null ) return root;
        if( root.val==key ){
            return insert( root.left , root.right );
        }
        dfs( root, key );
        return root;
    }
    public TreeNode dfs( TreeNode root, int key ){
        if( root == null ) return null;

        if( root.left!=null && root.left.val== key ){
            root.left = insert( root.left.left , root.left.right  );
            return null;
        }
        else if( root.right!=null && root.right.val== key ){
            root.right = insert( root.right.left , root.right.right );
            return null;
        }
        dfs( root.left , key  );
        dfs( root.right , key );
        return root;
    }
    public TreeNode insert( TreeNode left , TreeNode right ){
        if( left==null && right==null ) return null;
        else if( left==null ) return right;
        else if( right==null ) return left;

        if( right.left!=null ){
            insert( left , right.left );
        }
        else{
            right.left = left;
        }
        return right;
    }
}

改进

看了leetcode官解后，发现自己在dfs函数中没有利用到二叉搜索树的性质，即子节点的val不为key时，应通过val与key的大小关系，决定递归左子树还是右子树。