0
点赞
收藏
分享

微信扫一扫

制作Android分区镜像

Android开发指南 2024-05-06 阅读 42

使用dfs深度搜索,同值子树,要满足三个条件:

对于当前节点node,他的左子树血脉纯净(为同值子树),右子树血脉纯净(为同值子树),node的值等于左右子树节点的值。

全是if判断,推理!!!

class Solution:
    def countUnivalSubtrees(self, root: Optional[TreeNode]) -> int:
        n, b = self.dfs(root)
        return n
    
    def dfs(self, root):
        if not root: return 0, True
        n = 0
        b = True
        n1, b1 = self.dfs(root.left)
        n2, b2 = self.dfs(root.right)
        n = n1 + n2
        if not b1 or not b2:
            b = False
        if root.left and root.left.val != root.val:
            b = False
        if root.right and root.right.val != root.val:
            b = False
        if b: n += 1
        return n, b

使用dfs, 返回以root为根的所以节点的总和,节点数量。

没有任何技巧,全是感情!!!

class Solution:
    def __init__(self):
        self.m = 0

    def maximumAverageSubtree(self, root: Optional[TreeNode]) -> float:
        self.dfs(root)
        return self.m

    def dfs(self, root):
        # 返回以root为根的所以节点的总和,节点数量
        if not root: return 0, 0
        s1, c1 = self.dfs(root.left)
        s2, c2 = self.dfs(root.right)
        s = s1 + s2 + root.val
        c = c1 + c2 + 1
        self.m = max(self.m, s/c)
        return s, c

 可以把题目分成三个问题,使用三个dfs解决,可以发现左边界和右边界很相似,dfs传入一个idx判断是先从左走还是先从右走,另外题目说:根节点 不是 叶节点。但是数据中存在只有一个节点的情况需要注意。

class Solution:
    def __init__(self):
        self.leaf = []

    def boundaryOfBinaryTree(self, root: Optional[TreeNode]) -> List[int]:
        if not root.left and not root.right: return [root.val]
        ans = []
        if root.left:
            l = self.find_ls(root, 0)
            ans += l
        else:
            ans = [root.val]
        self.find_leaf(root)
        ans += self.leaf
        if root.right:
            r = self.find_ls(root, 1)
            ans += r[::-1]
            ans.pop()
        return ans

    def find_ls(self, root, idx):
        ans = [root.val]
        if idx == 1:
            root.left, root.right = root.right, root.left
        if root.left:
            ans += self.find_ls(root.left, idx)
        elif root.right:
            ans += self.find_ls(root.right, idx)
        else:
            return []
        return ans

    def find_leaf(self, root):
        if root.left:
            self.find_leaf(root.left)
        if root.right:
            self.find_leaf(root.right)
        if not root.left and not root.right:
            self.leaf.append(root.val)

任然使用dfs深度搜索,记录每一层的位置,然后在ans相应位置中插入

class Solution:
    def __init__(self):
        self.length = 0
        self.ans = []

    def findLeaves(self, root: Optional[TreeNode]) -> List[List[int]]:
        self.dfs(root)
        return self.ans

    def dfs(self, root):
        if not root: return 0
        n1 = self.dfs(root.left)
        n2 = self.dfs(root.right)
        n = max(n1, n2)
        if self.length - 1 < n:
            self.length += 1
            self.ans.append([])
        self.ans[n].append(root.val)
        return n + 1

还能补充知识吗!!!我的大脑🧠

举报

相关推荐

0 条评论