平衡二叉树

题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。

分析
首先我们要知道AVL树性质,不了解的可以看这篇文章:平衡二叉树(AVL)的理解和实现(Java)

我们知道平衡二叉树是高度平衡化的二叉搜索树,那么对于任意一个节点,其左右子树的高度差不大于1,如果存在某个节点左右子树深度差大于1,那么一定不是AVL树。

节点类:

public class TreeNode {
	int val = 0;
	TreeNode left = null;
	TreeNode right = null;

	public TreeNode(int val) {
		this.val = val;
	}
}

方法一

	/**
	 * 方法一
	 * @param root
	 * @return
	 */
	public boolean IsBalanced_Solution(TreeNode root) {
		if(root == null) {
			return true;
		}
		return Math.abs(depth(root.left) - depth(root.right)) <= 1 && IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
	}

	private int depth(TreeNode node) {
		if(node == null) return 0;

		int m = depth(node.left);
		int n = depth(node.right);

		return m > n ? m + 1 : n + 1;
	}

这是最直接的做法,遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。

但是我们发现该方法在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。

方法二

	/**
	 * 方法二
	 * @param root
	 * @return
	 */
	public boolean isAVL = true;
	
	public boolean IsBalanced_Solution_2(TreeNode root) {
		getdepth(root);
		return isAVL;
	}
	
	private int getdepth(TreeNode node) {
		if(!isAVL) return 0;
		if(node == null) return 0;
		int m = getdepth(node.left);
		int n = getdepth(node.right);
		if(Math.abs(m - n) > 1) {
			isAVL=false;
		}

		return m > n ? m + 1 : n + 1;
	}
posted @ 2019-03-28 21:51  如是说  阅读(293)  评论(0编辑  收藏  举报