题目:输入一棵二叉树,判断该二叉树是否是平衡二叉树。
结果:牛客网编译通过
代码及思路:
//中心思想:递归方式后续遍历树,只遍历一遍,遍历到某一节点的时候返回当前节点的深度,然后判断左子树右子树的深度差是否小于等于1,如果大于1则不为平衡二叉树
bool result = true; //定义全局变量存要返回的结果,初始化为真bool IsBalanced_Solution(TreeNode* pRoot) {
depth(pRoot);
return result;
}
int depth(TreeNode* root){ //后序遍历的方式返回当前节点的深度
if(root == NULL)
return 0;
int l = depth(root->left);
int r = depth(root->right);
if(abs(l-r)>1) result = false; //如果当前节点不是平衡二叉树,就把要返回的结果置为假
return l>r?l+1:r+1; //当前节点的深度等于左右子节点深度的最大值加一
}