码迷,mamicode.com
首页 > 其他好文 > 详细

[leetcode]450. Delete Node in a BST二叉搜索树删除节点

时间:2018-01-29 21:22:32      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:nbsp   ret   log   body   code   节点   眼睛   color   左右子树   

二叉树变量只是一个地址

public static void main(String[] args) {
        TreeNode t = new TreeNode(3);
        help(t);
        System.out.println(t.val);
    }
    public static void help(TreeNode t)
    {
        t.val = 6;
    }

上边代码通过地址改变了二叉树,输出为6,但是下边代码却只是传入函数的二叉树变量指向了另一个地址,外界的二叉树变量和二叉树的值没有变,输出还是3

public static void main(String[] args) {
        TreeNode t = new TreeNode(3);
        help(t);
        System.out.println(t.val);
    }
    public static void help(TreeNode t)
    {
        t = new TreeNode6);
    }

所以想改变二叉树,不能改变二叉树变量,而应该通过二叉树变量t调用val,left,right进行赋值,就可以改变,直接改变t只是让t指向另一课树,原本的树没有改变。

下边是答案,思路是先找到节点,再根据节点的不同情况进行操作。

最后的操作很乱,自己都看不下去了,应该递归的改变左右子树,但是眼睛太累了,有空再改吧。

public TreeNode deleteNode(TreeNode root, int key) {
        if(root==null) return null;
        if(root.val==key)
        {
            if (root.left==null)  return root.right;
            if (root.right==null) return root.left;
            TreeNode temp = root.right;
            while (temp.left!=null) temp = temp.left;
            if (root.left.right!=null) temp.left = root.left.right;
            root.left.right = root.right;
            root.val = root.left.val;
            if (root.left.left==null)
            {
                root.right = root.left.right;
                root.left = null;
            }
            else {
                //这里注意,由于两句话都要用到root.left,所以root.left最后再变
                root.right = root.left.right;
                root.left = root.left.left;
            }
        }
        else
        {
            if(root.val>key) root.left = deleteNode(root.left,key);
            else root.right = deleteNode(root.right,key);
        }
        return root;
    }

 

[leetcode]450. Delete Node in a BST二叉搜索树删除节点

标签:nbsp   ret   log   body   code   节点   眼睛   color   左右子树   

原文地址:https://www.cnblogs.com/stAr-1/p/8379335.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!