清风Python力扣算法刷题

力扣每日一题:872.叶子相似的树

2021-05-10  本文已影响0人  清风Python

872.叶子相似的树

https://leetcode-cn.com/problems/leaf-similar-trees/

难度:简单

题目:

请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

举个例子,如上图所示,给定一棵叶值序列为(6, 7, 4, 9, 8)的树。

如果有两棵二叉树的叶值序列是相同,那么我们就认为它们是叶相似的。

如果给定的两个根结点分别为root1 和root2的树是叶相似的,则返回true;否则返回 false 。

示例:

示例 1:

输入:root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
输出:true

示例 2:

输入:root1 = [1], root2 = [1]
输出:true

示例 3:

输入:root1 = [1], root2 = [2]
输出:false

示例 4:

输入:root1 = [1,2], root2 = [2,2]
输出:true

示例 5:

输入:root1 = [1,2,3], root2 = [1,3,2]
输出:false

分析

二叉树搜索叶子结点,要求我们对深度优先的方法具备相关掌握即可完成解题。
在深度优先搜索的过程中,我们总是先搜索当前节点的左子节点,再搜索当前节点的右子节点。
这里由于要搜索两个二叉树的叶子结点,所以单独抽离出dfs方法,进行调用后判断结果

解题:

# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right

class Solution:
    def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        def dfs(root, ret):
            if not root.left and not root.right:
                ret.append(root.val)
                return
            if root.left:
                dfs(root.left, ret)
            if root.right:
                dfs(root.right, ret)

        ret1 = []
        dfs(root1, ret1)
        ret2 = []
        dfs(root2, ret2)
        return ret1 == ret2

欢迎关注我的公众号: 清风Python,带你每日学习Python算法刷题的同时,了解更多python小知识。

我的个人博客:https://qingfengpython.cn

力扣解题合集:https://github.com/BreezePython/AlgorithmMarkdown

上一篇下一篇

猜你喜欢

热点阅读