leetcode DFS 938

最新推荐文章于 2024-10-30 15:22:37 发布

阿泽学习日记

最新推荐文章于 2024-10-30 15:22:37 发布

阅读量77

点赞数

文章标签：机器学习概率论人工智能

本文链接：https://blog.csdn.net/nendie1/article/details/123320892

版权

938

设置判断层次的数量

# 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 rangeSumBST(self, root: TreeNode, low: int, high: int) -> int:
        res = 0
        queue = []
        queue.append(root)
        while len(queue) > 0: #判断是否遍历完
            size = len(queue) #判断该层次的数量
            while size > 0: #遍历该层次的数量
                temp = queue.pop()
                if temp.val <= high and temp.val >=low:
                    res += temp.val
                if temp.left:
                    queue.append(temp.left)
                if temp.right:
                    queue.append(temp.right)
                size -= 1
        return res