872. 叶子相似的树(JavaScript)

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

举个例子,如上图所示,给定一颗叶值序列为 (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;
}

猜你喜欢

转载自blog.csdn.net/romeo12334/article/details/81456229