请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
提示:
- 给定的两颗树可能会有
1
到100
个结点。
思路:
首先要求得两个树的叶子序列,为避免重复,写一个新的函数:getLeaves,通过根节点获取树的叶子序列。
然后比较两个叶子序列是否相等。
JavaScript中,数组是以引用的方式存在的,不能通过a === b 来判断数组a与数组b是否相等。因此必须判断 a与b中的每一个元素是否都相等。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root1
* @param {TreeNode} root2
* @return {boolean}
*/
var leafSimilar = function(root1, root2) {
var leaves1 = getLeaves(root1),
leaves2 = getLeaves(root2);
if (leaves1.length !== leaves2.length){
return false;
}else {
for (var i in leaves1) {
if (leaves1[i] !== leaves2[i]) {
return false;
}
}
return true;
}
};
/**
* @param {TreeNode} root
* @return {List[]}
*/
var getLeaves = function(root) {
var leaves = [];
if (!root) {
return leaves;
}
if (root.left !== null) {
leaves = leaves.concat(getLeaves(root.left));
}
if (root.right !== null) {
leaves = leaves.concat(getLeaves(root.right));
}
if (root.left === null && root.right === null) {
leaves.push(root.val);
}
return leaves;
}