您的位置:首页 > 编程语言 > Java开发

【LintCode-93】平衡二叉树(Java实现-递归算法)

2017-11-05 23:37 295 查看


第一次代码记录:

public boolean isBalanced(TreeNode root) {
if(root==null||(root.left==null&&root.right==null)){
return true;
}
if(root.left!=null&&root.right!=null){
return isBalanced(root.left)&&isBalanced(root.right);
}else if(root.left!=null){
return depth(root.left)>1?false:true;
}else{
return depth(root.right)>1?false:true;
}
// write your code here
}
public int depth(TreeNode root){
if(root==null){
return 0;
}
int leftRootDepth=depth(root.left);
int    rightRootDepth=depth(root.right);
return (leftRootDepth>rightRootDepth?leftRootDepth:rightRootDepth)+1;
}


LintCode测试BUG指出:

在LintCode提交测试,显示正常通过,但其实是不全对,因为代码里只判断了单独的左右子树是否平衡(return isBalanced(root.left)&&isBalanced(root.right);),而没有判断当前这个节点,左右子树的高度差是否满足小于等于1的平衡条件。测试数据:{1,2,3,4,5,#,#,6,7,8,9,#,#,#,#} 根节点左右子树不平衡。

第二次代码记录:

public boolean isBalanced(TreeNode root) {
if(root==null){
return true;
}
int depthRootLeft=depth(root.left);
int depthRootRight=depth(root.right);
if(Math.abs(depthRootLeft-depthRootRight)<=1){
return isBalanced(root.left)&&isBalanced(root.right);
}
return false;
}
public int depth(TreeNode root){
if(root==null){
return 0;
}
int depthLeft=depth(root.left);
int depthRight=depth(root.right);
return (depthLeft>depthRight?depthLeft:depthRight)+1;//取左右子树最大值
b3dc
+1;
}


个人解题思路:

判断当前这个二叉树是否平衡,其实就是判断这个二叉树上的没有treeNode节点是否平衡。

1.我们需要先求出这个左右子树的高度(通过depth()),depth()里用了递归算法求出其参数root节点的左右子树高度,并取其最大值+1。

2.回来isBalanced()里,取到左右子树高度后,进行判断当前节点是否平衡,如果平衡就继续递归其左右子树节点是否平衡,并将两者结果用且(&&)返回。

如果您有更好的解法,欢迎您在留言区补充,感谢!!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: