检查二叉树是否平衡

时间:2016-02-23 19:29:47

标签: c++ algorithm height binary-tree balance

我在下面实现了C ++代码片段,以检查二叉树是否平衡,即左右子树的高度最多相差1.但是,我不确定它是否有效,或者反复检查子树以糟糕的方式。有人可以指导我吗?

 unordered_map <Node*, int>  height;
    struct Node{
       int key;
       Node* left;
       Node* right;
    }
    bool isBalanced(Node* root){
        if (root == nullptr){
             height[root] = -1;
            return true;
        }
        if (!isBalanced(root->left)) return false;
        if (!isBalanced(root->right)) return false;

        height[root] = max(height[root->left], height[root->right]) + 1;


        return (abs(height[root->left] - height[root->right]) < 1);
}

1 个答案:

答案 0 :(得分:3)

我会尝试使用伪代码传递这个想法。

int CheckTreeHeight(Node root)
{
  if(root == null) return 0; // Height of 0.

  // Check if left is balanaced
  int leftChildHeight = CheckTreeHeight(root.left);
  if(leftChildHeight == -1) return -1; // Not Balanced

  // Check if right is balanaced
  int rightChildHeight = CheckTreeHeight(root.right);
  if(rightChildHeight == -1) return -1; // Not Balanced

  // Check if current node is balanced
  int heightDifference = leftChildHeight - rightChildHeight;

  if(Math.abs(heightDifference) > 1)
   return -1; // not balanaced
  else
   return Math.max(leftChildHeight, rightChildHeight) + 1; // Return Height
}

bool IsBalanced(Node root)
{
   if(CheckTreeHeight(root) == -1)
   {
      return false;
   }
   else
   {
      return true;
   }
}

此算法执行O(n)时间复杂度和O(H)空间复杂度,其中h是树的高度。

正如算法作者所提到的,我们的想法是我们以递归方式检查根的子节点(即leftright),直到找到不平衡的节点,然后返回{{1} }。

使用这种技术,如果任何子树不平衡,我们会立即返回。

有关此实施的更多信息,请参阅参考文献中提到的书。

<强>参考
Cracking the Coding Interview 6th Edition