872. Leaf-Similar Trees - LeetCode

Question

872. Leaf-Similar Trees

Solution

题目大意：

如果两个二叉树的叶子节点相同就认为这两个二叉树相似。给两个二叉树判断是否相似。

思路：

用递归把两个二叉树的叶子节点遍历出来，再比较叶子节点是否相同

Java实现：

public boolean leafSimilar(TreeNode root1, TreeNode root2) {
    // 把叶子节点遍历出来，比较
    List<Integer> leftLeafs = new ArrayList<>();
    List<Integer> rightLeafs = new ArrayList<>();
    initLeaf(root1, leftLeafs);
    initLeaf(root2, rightLeafs);
    if (leftLeafs.size() != rightLeafs.size()) return false;
    for (int i = 0; i < leftLeafs.size(); i++) {
        if (leftLeafs.get(i) != rightLeafs.get(i)) return false;
    }
    return true;
}

public void initLeaf(TreeNode root, List<Integer> leafs) {
    if (root.left == null &&  root.right == null) leafs.add(root.val);
    if (root.left != null) initLeaf(root.left, leafs);
    if (root.right != null) initLeaf(root.right, leafs);
}