检查二叉树是否平衡

《检查二叉树是否平衡》实现一个函数,检查二叉树是否平衡。在这个问题中,平衡树的定义如下:任意一个结点,其两棵子树的高度差不超过1.

              直接递归访问整棵树,计算每个结点两棵子树的高度。

public static int getHeight(TreeNode root)
{
if(root == null) return 0;//终止条件
return Math.max(getHeight(root.left), getHeight(root.right))+1;
}
public static boolean isBalanced(TreeNode root)
{
if(root == null) return true;//终止条件
int heightDiff=getHeight(root.left)-getHeight(root.right);
if(Math.abs(heightDiff)>1)
return false;
else//递归
{
return isBalanced(root.left) && isBalanced(root.right); 

}
}

《检查二叉树是否平衡》

public static int checkHeight(TreeNode root )
{
if(root == null)
{
return 0;//高度为0
}
//检查左子树是否平衡
int leftHeight=checkHeight(root.left);
if(leftHeight==-1)
{
return -1;
}
//检查右子树是否平衡
int rightHeight=checkHeight(root.right);
if(rightHeight == -1)
{
return -1;
}

/*检查当前结点是否平衡*/

int heightDiff=leftHeight-rightHeight;

if(Math.abs(heightDiff)>1)

      return -1;

else
{
//返回高度
     return Math.max(leftHeight, rightHeight)+1;
}
}

public static boolean isBalanced(TreeNode root)
{
if(checkHeight(root)== -1)
{
return false;
}
else
{
return true;
}
}

这段代码需要O(N)的时间和O(H)的空间,H是树的高度。

    原文作者:平衡二叉树
    原文地址: https://blog.csdn.net/wangfengfan1/article/details/47843673
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞