一刷剑指offer（39）——二叉树的深度

题目：
输入一颗二叉树的根结点，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。
二叉树结点定义如下:
struct BinaryTreeNode
{
    int    m_nValue;
    BinaryTreeNode*    m_pLeft;
    BinaryTreeNode*    m_pRight;
};

递归：

如果一棵树只有一个结点，它的深度为1。

如果根结点只有左子树而没有右子树，那么树的深度是其左子树深度+1；

如果根结点只有右子树而没有左子树，那么树的深度是其右子树深度+1；

如果既有左子树又有右子树，那么树的深度是max{左子树深度，右子树深度}+1。

int TreeDepth(BinaryTreeNode* pRoot)
{
    if(pRoot==NULL)
        return 0;
    int nLeft=TreeDepth(pRoot->m_pLeft);
    int nRight=TreeDepth(pRoot->m_pRight);
    return (nLeft>nRight)?(nLeft+1):(nRight+1);
}

 

题目：
输入一颗二叉树的根结点，判断该树是否为平衡二叉树。如果某二叉树中任意结点的左右子树深度相差不超过1，那么它就是一颗平衡二叉树。

最直接的方法就是遍历树的每个结点时，都用一次TreeDepth函数得到左右子树的深度。虽然简单，但是由于每个结点会被重复遍历多次，因此效率不高。

bool IsBalanced(BinaryTreeNode* pRoot)
{
    if(pRoot==NULL)
        return true;
    int left=TreeDepth(pRoot->m_pLeft);
    int right=TreeDepth(pRoot->m_pRight);
    int diff=left-right;
    if(diff>1 || diff<-1)
        return false;
    return IsBalanced(pRoot->m_pLeft) && IsBalanced(pRoot->m_pRight);
}

如果用后序遍历的方式遍历二叉树的每一个结点，在遍历到一个结点之前我们就已经遍历了它的左右子树。只要在遍历每个结点的时候记录它的深度，就可以一边遍历一边判断是否平衡。

bool IsBalanced(BinaryTreeNode* pRoot,int* pDepth)
{
    if(pRoot==NULL)
    {
        *pDepth=0;
        return true;
    }
    int left,right;
    if(IsBalanced(pRoot->m_pLeft,&left) && IsBalanced(pRoot->m_pRight,&right))
    {
        int diff=left-right;
        if(diff<=1 && diff>=-1)
        {
            *pDepth=1+(left>right?left:right);
             return true;
        }
    }
    return false;
}