zoukankan      html  css  js  c++  java
  • 算法(从二叉搜索数中删除一个节点)

    给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
    
    一般来说,删除节点可分为两个步骤:
    
    首先找到需要删除的节点;
    如果找到了,删除它。

    说明: 要求算法时间复杂度为 O(h),h 为树的高度。

    
    

    示例:

    
    
    root = [5,3,6,2,4,null,7]
    key = 3
    
        5
       / 
      3   6
     /    
    2   4   7
    
    给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
    
    一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
    
        5
       / 
      4   6
     /     
    2       7
    
    另一个正确答案是 [5,2,6,null,4,null,7]。
    
        5
       / 
      2   6
          
        4   7
     
    /**
     * Definition for a binary tree node.
     * type TreeNode struct {
     *     Val int
     *     Left *TreeNode
     *     Right *TreeNode
     * }
     */
    func deleteNode(root *TreeNode, key int) *TreeNode {
        if (root == nil) {
            return nil
        }
        if (key < root.Val) {
            root.Left = deleteNode(root.Left, key)
            return root
        } else if (key > root.Val) {
            root.Right = deleteNode(root.Right, key)
            return root
        } else {
            if (root.Left == nil) {
                return root.Right
            }
            if (root.Right == nil) {
                return root.Left
            }
            // new 和 make的区别,要记住
            //通过make创建对象 make只能创建slice  chan map
            //make 只能用来分配及初始化类型为 slice、map、chan 的数据。new 可以分配任意类型的数据;
            //new 分配返回的是指针,即类型 *Type。make 返回引用,即 Type;
            //new 分配的空间被清零。make 分配空间后,会进行初始化;
            //make 函数只用于 map,slice 和 channel,并且不返回指针
            node := new(TreeNode)
            node = findMin(root.Right)
            root.Val = node.Val
            root.Right = deleteNode(root.Right, node.Val)
        }
        return root
    }
    
    func findMin(node *TreeNode) *TreeNode {
        for {
            // golang 没有while
            if node.Left == nil {
                break
            }
            node = node.Left
        }
        return node
    }
  • 相关阅读:
    Spring AOP概念理解
    五分钟快速掌握RPC原理及实现
    Linux常用命令汇总
    一致性哈希算法原理
    RPC原理及实现
    IO设计模式:Reactor和Proactor对比
    到底什么时候该使用MQ?
    eclipse查看一个方法被谁引用(调用)的快捷键四种方式
    maven build pulgin
    VSCode 常用setiings.json设置
  • 原文地址:https://www.cnblogs.com/cjjjj/p/12550356.html
Copyright © 2011-2022 走看看