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

求二叉树的深度

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

输入:二叉树如下图

《剑指Offer——面试题39:二叉树的深度(如何判断某棵二叉树是否是平衡二叉树)》

输出:二叉树的深度为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;
}
    原文作者:平衡二叉树
    原文地址: https://blog.csdn.net/m0_37885286/article/details/80164088
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞