leetcode刷題. 面試題55 - II. 平衡二叉樹

輸入一棵二叉樹的根節點，判斷該樹是不是平衡二叉樹。如果某二叉樹中任意節點的左右子樹的深度相差不超過1，那麼它就是一棵平衡二叉樹。

 bool IsBalanced(TreeNode *root, int &depth) {

         if (nullptr == root)
                 return true;
         
         int left = 0, right = 0;

         if (IsBalanced(root->left, left) && IsBalanced(root->right, right)) {

                 depth = left > right ? left + 1 : right + 1;
                 int diff = abs(left - right);
                 if (diff > 1)
                         return false;
                 return true;
         }
         return false;
 }

bool isBalanced(TreeNode* root) {

    int depth = 0;
    return IsBalanced(root, depth);
}