剑指offer——平衡二叉树

    题目：输入一棵二叉树，判断该二叉树是否是平衡二叉树

    代码：

class Solution {
public:
     int TreeDepth(TreeNode* pRoot)
    {
        if(pRoot == NULL)
            return 0;
        
        int pLeft = TreeDepth(pRoot->left);
        int pRight = TreeDepth(pRoot->right);
        return ((pLeft > pRight)? pLeft+1 : pRight+1);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == NULL)
            return true;
        int pLeft = TreeDepth(pRoot->left);
        int pRight = TreeDepth(pRoot->right);
        int diff = pLeft -pRight;
        if(fabs(diff) > 1)  // 任意结点的左右子树的深度的差值大于 1 了
            return false;
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
    

    分析：判断一棵二叉树是不是平衡二叉树，需要判断该二叉树任意结点的左右子树的深度不超过1 。由于刚刚做完一题利用递归的方法求二叉树的深度，所以很容易写出来代码：在遍历结点的时候，同时计算该结点左右子树的深度，以及深度的差值，如果超过 1 ，就返回 false 。就这样，递归下去。

    这种方法虽然很简洁，但是效率不高。因为遍历一个结点的时候，要遍历它的子树，这样，遍历下一个结点的时候，显然会重复遍历一些结点。如果重复遍历结点，那么，就会影响时间性能。

    每个结点只被遍历一次的解法。

代码（对《剑指offer》书上 P210 页做了修改，使程序更完整、简洁）：

class Solution {
public:
    bool IsBalanced(TreeNode *root, int & dep){
        if(root == NULL){
            return true;
        }
        int left = 0;
        int right = 0;
        if(IsBalanced(root->left,left) && IsBalanced(root->right, right)){
            int dif = left - right;
            if(dif<-1 || dif >1)
                return false;
            dep = (left > right ? left : right) + 1;
            return true;
        }
        return false;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        int dep = 0;
        return IsBalanced(pRoot, dep);
    }
}；
    

分析：如果采用后序遍历的方式，我们再遍历一个结点的时候，它的左右子树就已经被遍历过了。只要在遍历结点的时候，记录它的深度，我们就可以一边遍历结点，一边遍历结点是不是平衡的。