平衡二叉树——js

平衡二叉树

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路

正常思路，应该会获得节点的左子树和右子树的高度，然后比较高度差是否小于1。
可是这样有一个问题，就是节点重复遍历，影响效率。所以采用另一种方法：在求高度的同时判断是否平衡，如果不平衡就返回-1，否则返回树的高度。并且当左子树高度为-1时，就没必要去求右子树的高度了，可以直接一路返回到最上层了

代码

//重复遍历了
function IsBalanced_Solution(pRoot) {
  if (pRoot == null) return true;
  let leftLen = TreeDepth(pRoot.left);
  let rightLen = TreeDepth(pRoot.right);
  return Math.abs(rightLen - leftLen) <= 1 && IsBalanced_Solution(pRoot.left) && IsBalanced_Solution(pRoot.right);
}
function TreeDepth(pRoot) {
  if (pRoot == null) return 0;
  let leftLen = TreeDepth(pRoot.left);
  let rightLen = TreeDepth(pRoot.right);
  return Math.max(leftLen, rightLen) + 1;
}
/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */

//同时判断
function IsBalanced_Solution(pRoot)
{
    // write code here
    return TreeDepth(pRoot) !==-1;
}
function TreeDepth(pRoot){
    if(pRoot === null) return 0;
    var left = TreeDepth(pRoot.left);
    if(left === -1)return -1;
    var right = TreeDepth(pRoot.right);
    if(right === -1)return -1;
    return Math.abs(left - right) > 1 ? -1: Math.max(left,right) + 1;
}

知识点

abs() 方法可返回数的绝对值。