算法训练营day20,二叉树9

type TreeNode struct {

  Val   int

  Left  *TreeNode

  Right *TreeNode

}

//669. 修剪二叉搜索树

//本题与450. 删除二叉搜索树中的节点 相似,不同之处在于本题删除不止一个节点,450不需要遍历全部二叉树,但本题需要全部遍历,而且要随时调整二叉树结构

func trimBST(root *TreeNode, low int, high int) *TreeNode {

  if root == nil {

    return root

  }

 //找到要删除的左子树

  if root.Val < low {

   //因为不确定左子树是否还要删除,因此要不断向右遍历,因为是二叉树搜索树,当前节点如果比low小,当前节点左子树一定比low小,因此直接舍弃,不用遍历

    right := trimBST(root.Right, low, high)

    return right

  }

  if root.Val > high {

 //同样因为不确定右子树是否还要删除,因此要不断向左遍历,因为是二叉树搜索树,当前节点如果比high 大,当前节点右子树一定比high 大,因此直接舍弃,不用遍历

    life := trimBST(root.Left, low, high)

    return life

  }

  root.Left = trimBST(root.Left, low, high)

  root.Right = trimBST(root.Right, low, high)

  return root

}

 //108. 将有序数组转换为二叉搜索树

//因为是有序数组,让中间节点作为根据节点,然后进行前序遍历中左右

func sortedArrayToBST(nums []int) *TreeNode {

  if len(nums) == 0 {

    return nil

  }

//寻找中间值索引

  mid := len(nums) / 2

  root := &TreeNode{Val: nums[mid]}  //中

  root.Left = sortedArrayToBST(nums[:mid]) //左

  root.Right = sortedArrayToBST(nums[mid+1:]) //右

  return root

}

//538. 把二叉搜索树转换为累加树

func convertBST(root *TreeNode) *TreeNode {

  if root == nil {

    return nil

  }

  prev := 0

  var traversal func(tree *TreeNode)

  traversal = func(tree *TreeNode) {

    if tree == nil {

      return

    }

    // 中序遍历是左中右,但我们要先取最大值,所以要逆中序遍历,即右中左

    traversal(tree.Right)      // 右

    tree.Val = tree.Val + prev //中

    prev = tree.Val            //把当前值传给前一个指针

    traversal(tree.Left)       //左

  }

  traversal(root)

  return root

}

  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值