剑指Offer——面试题39：二叉树的深度（如何判断某棵二叉树是否是平衡二叉树）

求二叉树的深度

---

题目：输入一棵二叉树的根结点，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。例如下图中的二叉树的深度为4，因为它从根结点到叶结点最长的路径包含4个结点（从根结点1开始，经过结点2和结点5，最终到达叶结点7）。

输入：二叉树如下图

输出：二叉树的深度为4

思路：1、利用递归的思想，对于每个节点，判断它的左子树深度为多少，判断它的右子树深度为多少
2、取左子树深度与左子树深度最大的+1，即为该节点的深度
Ps：用公式表示即：max[ (leftLen+1), (rightLen+1) ]

/*
实现：求二叉树深度的函数TreeDepth
*/
int TreeDepth(TreeNode* pRoot)
    {
        if(pRoot==NULL){
            return 0;
        }
        int leftLen = TreeDepth(pRoot->left);
        int rightLen= TreeDepth(pRoot->right);
        return leftLen>rightLen ? (leftLen+1) : (rightLen+1);
    }

判断是否为平衡二叉树

---

/*
实现：判断是否是平衡二叉树
*/
//简单版，但是存在不足，就是会遍历很多重复的节点
bool IsBalanced(BinaryTreeNode* pRoot){
    if (pRoot == NULL){
        return true;
    }
    int leftLen = TreeDepth(pRoot->left);
    int rightLen = TreeDepth(pRoot->right);
    int diff = leftLen - rightLen;
    if (diff > 1 || diff < -1){
        return false;
    }
    return IsBalanced(pRoot->left) && IsBalanced(pRoot->right);
}

//利用后序遍历，这样对每个节点只需要遍历一次
bool IsBalanced(BinaryTreeNode* pRoot,int* depth){
    if (pRoot == NULL){
        *depth = 0;
        return true;
    }
    if (IsBalanced(pRoot->left, &leftLen) && IsBalanced(pRoot->right, &rightLen)){
        int diff = leftLen - rightLen;
        if (diff <= 1 || diff >= -1){
            *depth = leftLen > rightLen ? leftLen + 1 : rightLen + 1;
            return true;
        }
    }
    return false;
}