标签:判断 表示 balance mat https turn 收获 语言 过程
本文参考自《剑指offer》一书,代码采用Java语言。
输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
在(55-1) 二叉树的深度基础上修改:计算树的深度,树的深度=max(左子树深度,右子树深度)+1。在遍历过程中,判断左右子树深度相差是否超过1,如果不平衡,则令树的深度=-1,用来表示树不平衡。最终根据树的深度是否等于-1来确定是否为平衡树。
测试算例
1.功能测试(左斜树、右斜树、平衡或者不平衡树)
3.特殊测试(一个结点,null)
//题目:输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中 //任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。 public class BalancedBinaryTree { public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } public boolean IsBalanced_Solution(TreeNode root) { return getDepth(root)!=-1; } public int getDepth(TreeNode root) { if(root==null) return 0; int left=getDepth(root.left); if(left==-1) return -1; int right=getDepth(root.right); if(right==-1) return -1; return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right); } /* //自己开始想的方法,但是一定要把树给遍历完才行;上面的方法实现了剪枝 boolean isBalanced=true; public boolean IsBalanced_Solution(TreeNode root) { TreeDepth(root); return isBalanced; } public int TreeDepth(TreeNode root) { if(root==null) return 0; int left=TreeDepth(root.left); int right=TreeDepth(root.right); if(left-right>1 || right-left>1) isBalanced=false; return Math.max(left+1,right+1); } */ }
1.在判断出树不平衡后,进行剪枝(即代码中直接返回-1,不再对其他子树进行判断),以提高效率。
标签:判断 表示 balance mat https turn 收获 语言 过程
原文地址:https://www.cnblogs.com/yongh/p/9959131.html