代码随想录算法训练营0529| 669. 修剪二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树

669. 修剪二叉搜索树

跳过难题

感觉忙起来根本没有闲心思做更深的思考,只是机械的看懂了,刷完了

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

构造二叉搜索树:就是顺序把数据存入二叉树,不断使用二分法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        root = self.input_num(nums,0,len(nums)-1)#左闭右闭
        return root

    def input_num(self,nums,left,right):
        if right < left:#终止条件
            return None
## 每层要做的事:把遍历到的这个节点转化为treenode
        mid = (left+right)//2
        # print(mid)
        # print(left)
        node = TreeNode(nums[mid])

        node.left = self.input_num(nums,left,mid-1)#左
        node.right = self.input_num(nums,mid+1,right)#

        return node

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

思路很妙,因为是大于等于本身的数的合集,那么按照右中左的顺序遍历,每次只需要加上当前节点的值,那么此刻的和即为本节点需要存储的值。

递归的写法,假设推到最后一层(第一次做的运算),整理需要做的操作

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def convertBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        self.curSum = 0#存储求和的值
        self.getSumNum(root)
        return root


    def getSumNum(self,node):
        if not node:
            return

        self.getSumNum(node.right)#追到最深的右节点

        self.curSum += node.val#当前节点要存储的值
        node.val = self.curSum#更改当前节点的值

        self.getSumNum(node.left)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值