请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个叶值序列 。
举个例子,如上图所示,给定一棵叶值序列为 (6, 7, 4, 9, 8) 的树。
如果有两棵二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个根结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。
示例 1:
输入:root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
输出:true
示例 2:
输入:root1 = [1], root2 = [1]
输出:true
示例 3:
输入:root1 = [1], root2 = [2]
输出:false
示例 4:
输入:root1 = [1,2], root2 = [2,2]
输出:true
示例 5:
输入:root1 = [1,2,3], root2 = [1,3,2]
输出:false
提示:
- 给定的两棵树可能会有 1 到 200 个结点。
- 给定的两棵树上的值介于 0 到 200 之间。
解答
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> root1Leaves = new ArrayList<>();
List<Integer> root2Leaves = new ArrayList<>();
getLeavesList(root1, root1Leaves);
getLeavesList(root2, root2Leaves);
if (root1Leaves.size() != root2Leaves.size()) return false;
for (int i = 0; i < root1Leaves.size(); i++)
if (!root1Leaves.get(i).equals(root2Leaves.get(i))) return false;
return true;
}
// 序列化树,遍历树时将叶节点维护到leavesList中
private void getLeavesList(TreeNode node, List<Integer> leavesList) {
if (node == null) return;
if (node.left == null && node.right == null) {
leavesList.add(node.val);
return;
}
getLeavesList(node.left, leavesList);
getLeavesList(node.right, leavesList);
}
}