剑指offer（39）平衡二叉树

题目描述

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

 

题目分析

第一种方法：

　　正常思路，应该会获得节点的左子树和右子树的高度，然后比较高度差是否小于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 IsBalancedSolution(pRoot) {
  return TreeDepth(pRoot) !== -1;
}
function TreeDepth(pRoot) {
  if (pRoot === null) return 0;
  const leftLen = TreeDepth(pRoot.left);
  if (leftLen === -1) return -1;
  const rightLen = TreeDepth(pRoot.right);
  if (rightLen === -1) return -1;
  return Math.abs(leftLen - rightLen) > 1 ? -1 : Math.max(leftLen, rightLen) + 1;
}