面试题55(2):平衡二叉树

题目

输入一颗二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一颗平衡二叉树。例如下图就是一颗平衡二叉树

image.png

解题思路

1. 用后序遍历的方式遍历二叉树的每个节点，那么在遍历到一个节点之前就已经遍历了它的左、右子树。
2. 只要在遍历每个节点的时候记录它的深度(某一节点的深度等于它到叶节点的路径的长度)，就可以一边遍历一边判断每个节点是不是平衡的。

代码

class Solution{
    public:
        bool IsBalanced(BinaryTreeNode* pRoot)
        {
            int depth = 0;
            return IsBalanced(pRoot,&depth);
        }
        bool IsBalanced(BinaryTreeNode* pRoot,int *pDepth)
        {
            if(pRoot == NULL)
            {
                *pDepth = 0;
                return true;
            }

            int left,right;
            //cout << "left: " << *left << "right: " << *right << endl;
            if(IsBalanced(pRoot->left,&left) && IsBalanced(pRoot->right,&right))
            {
                int diff = left - right;
                if(diff<=1 && diff>=-1)
                {
                    *pDepth = 1+ (left>right?left:right);
                    return true;
                }
            }

            cout << "left: " << left << "right: " << right << endl;
            return false;
        }
};

完整代码见Github