剑指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);
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· 新年开篇:在本地部署DeepSeek大模型实现联网增强的AI应用
· 程序员常用高效实用工具推荐,办公效率提升利器!
· Janus Pro:DeepSeek 开源革新,多模态 AI 的未来
· 【译】WinForms:分析一下(我用 Visual Basic 写的)