Leetcode 872. 叶子相似的树

题目链接

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

题目描述

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

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

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

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

提示：

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

题解

把每棵树的叶子节点存起来，在进行比较。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
        if ((root1 != null && root2 != null) || (root1 == null && root2 == null)) {
            List<Integer> list1 = new ArrayList<>();
            List<Integer> list2 = new ArrayList<>();
            findLeaf(root1, list1);
            findLeaf(root2, list2);
            if (list1.size() != list2.size()) { return false; }
            for (int i = 0; i < list1.size(); i++) {
                if (list1.get(i) != list2.get(i)) { return false; }
            }
            return true;
        }
        return false;
    }
    public void findLeaf(TreeNode root, List<Integer> list) {
        if (root == null) { return; }
        if (root.left == null && root.right == null) { list.add(root.val); }
        findLeaf(root.left, list);
        findLeaf(root.right, list);
    }
}