872. 叶子相似的树 - 力扣（LeetCode）

下面是一个完整的 Python 函数，接收两个二叉树的根节点 root1 和 root2，返回它们是否叶相似。

代码实现

class TreeNode:def __init__(self, val=0, left=None, right=None):self.val = valself.left = leftself.right = rightdef leafSimilar(root1, root2):def getLeafSequence(root):leaf_values = []def dfs(node):if not node:returnif not node.left and not node.right:  # 叶子节点leaf_values.append(node.val)dfs(node.left)dfs(node.right)dfs(root)return leaf_valuesreturn getLeafSequence(root1) == getLeafSequence(root2)

函数说明

• leafSimilar(root1, root2):
  • 通过 getLeafSequence(root) 获取两棵树的叶子序列。
  • 比较两个序列是否相同，并返回 True 或 False。

这个函数直接传入两个二叉树的根节点即可判断它们是否叶相似。