LeetCode872-叶子相似的树

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/leaf-similar-trees/

中文链接：https://leetcode-cn.com/problems/leaf-similar-trees/

题目详述

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

img

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

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

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

提示：

  • 给定的两颗树可能会有 1 到 100 个结点。

题目详解

分别求出两棵树的叶值序列，然后再比较叶值序列是否相等。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public class LeetCode_00872 {

    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
        List<Integer> list1 = new ArrayList<>();
        List<Integer> list2 = new ArrayList<>();
        getLeafList(root1, list1);
        getLeafList(root2, list2);
        return list1.size() == list2.size() && list1.equals(list2);
    }

    private void getLeafList(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        if (root.left == null && root.right == null) {
            list.add(root.val);
        }
        getLeafList(root.left, list);
        getLeafList(root.right, list);
    }
}