请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
思路:深度优先搜索,递归,找到叶子节点就把其值加入到集合,没有左右子树的节点就是叶子节点,然后比较两个集合是不是一样。
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> leavesList1 = new ArrayList<>();
List<Integer> leavesList2 = new ArrayList<>();
dfs(root1,leavesList1);
dfs(root2,leavesList2);
return leavesList1.equals(leavesList2);
}
private void dfs(TreeNode node,List<Integer> leavesValue){
if (node != null){
if (node.left == null && node.right == null)
leavesValue.add(node.val);
dfs(node.left,leavesValue);
dfs(node.right,leavesValue);
}
}