判断一棵树是不是平衡二叉树

最新推荐文章于 2022-03-23 10:03:07 发布

wwzheng16

最新推荐文章于 2022-03-23 10:03:07 发布

阅读量396

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/wwzheng16/article/details/81065827

数据结构专栏收录该内容

45 篇文章

订阅专栏

本文介绍了一种用于检测二叉树是否为平衡二叉树的算法。通过递归计算每个节点的左右子树深度，并判断差值是否大于1来确定整棵树的平衡状态。示例代码展示了创建二叉树及调用检测函数的过程。

import org.junit.Test;

public class solution {
   @Test
   public void testFunc(){
   TreeNode root1 = new TreeNode(1);
   TreeNode root2 = new TreeNode(2);
   TreeNode root3 = new TreeNode(3);
   TreeNode root4 = new TreeNode(4);
   TreeNode root5 = new TreeNode(5);
   TreeNode root6 = new TreeNode(6);
   root1.left=root2;
   root1.right=root3;
   root2.left=root4;
   root2.right=root5;
   root5.left=root6;
   boolean balace = isBalance(root1);
   System.out.println("res: "+balace);
   }
//   判断一棵树是不是平衡二叉树
   public boolean isBalance(TreeNode root){
       if (root==null) {
           return true;
       }
       int diff = Math.abs(rootDepth(root.left)-rootDepth(root.right));
       if (diff<-1 || diff>1) {
           return false;
       }
       return isBalance(root.left)&&isBalance(root.right);

   }
   public int rootDepth(TreeNode root){
       if (root==null) {
           return 0;
       }
       return Math.max(rootDepth(root.left), rootDepth(root.right))+1;
   }
}