872. 叶子相似的树-每日一题

一、题目

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

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

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

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

点击查看原题

二、思路

       需要对叶子节点从左到右依次记录其数值，如果记录完全一致，那么就是叶相似，数量或者数值不一致，那就不是叶相似。
       比较经典且容易实现的，就是使用中序遍历的方式，中序遍历保证了可以依次从左往右访问叶子节点，使用两个List记录遍历过的节点，进行对比List的长度和每个元素数值，就可以实现本题要求。

三、代码

class Solution {
    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
        List<Integer> memo1 = new ArrayList();
        List<Integer> memo2 = new ArrayList();
        inorder(root1, memo1);
        inorder(root2, memo2);
        if (memo1.size() != memo2.size()) { // 如果叶子节点数量不一致，不是叶相似
            return false;
        }
        for (int i = 0; i < memo1.size(); i++) {
            if (memo1.get(i) != memo2.get(i)) {	// 有任何一个元素不相等，则不是叶相似
                return false;
            }
        }
        return true;
    }
    // 中序遍历从左往右依次访问叶子节点
    public void inorder(TreeNode root, List<Integer> memo) {
        if (root == null) {
            return ;
        }
        inorder(root.left, memo);	// 先访问左孩子
        if (root.left == null && root.right == null) {	// 当叶子时记录值
            memo.add(root.val);
        }
        inorder(root.right, memo);
    }
}

       m、n分别为两棵树的节点规模，时间复杂度为O(m+n)，空间复杂度为O(m+n)。