Processing math: 100%

剑指offer系列——39.平衡二叉树

Q:输入一棵二叉树,判断该二叉树是否是平衡二叉树。
A:
结合上一题的计算树的高度。

    bool IsBalanced_Solution(TreeNode *pRoot) {
        vector<int> diff = {-1, 0, 1};
        if (pRoot == nullptr)
            return true;
        int l = TreeDepth(pRoot->left);
        int r = TreeDepth(pRoot->right);
        int d = l - r;
        vector<int>::iterator it;
        it = find(diff.begin(), diff.end(), d);
        if (it == diff.end())
            return false;
        else {
            bool left = IsBalanced_Solution(pRoot->left);
            bool right = IsBalanced_Solution(pRoot->right);
            return left && right;
        }
    }

    int TreeDepth(TreeNode *pRoot) {
        if (pRoot == nullptr)
            return 0;
        int l = TreeDepth(pRoot->left);
        int r = TreeDepth(pRoot->right);
        return l > r ? l + 1 : r + 1;
    }

但这样做有一个问题,就是下层一直累积遍历很多次。这样直接在遍历过程中进行判断。

    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);
    }
posted @   Shaw_喆宇  阅读(102)  评论(0编辑  收藏  举报
编辑推荐:
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
阅读排行:
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· 新年开篇:在本地部署DeepSeek大模型实现联网增强的AI应用
· 程序员常用高效实用工具推荐,办公效率提升利器!
· Janus Pro:DeepSeek 开源革新,多模态 AI 的未来
· 【译】WinForms:分析一下(我用 Visual Basic 写的)
点击右上角即可分享
微信分享提示