平衡二叉树 java

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路：

1、重复遍历结点

参考上一题求二叉树的深度，先求出根结点的左右子树的深度，然后判断它们的深度相差不超过1，如果否，则不是一棵二叉树；如果是，再用同样的方法分别判断左子树和右子树是否为平衡二叉树，如果都是，则这就是一棵平衡二叉树。

但上面的方法在判断子树是否为平衡二叉树时，会重复遍历树的结点，不断地求子树的深度，所以效率不高。

2、遍历一遍结点

我们在遍历结点的同时记录下该结点的深度，这样就可以避免了重复访问。

public boolean IsBalanced_Solution(TreeNode root) {

		if (root == null) {
			return true;
		}

		int leftDepth = TreeDepth(root.left);
		int rightDepth = TreeDepth(root.right);

		int diff = leftDepth - rightDepth;
		if (diff > 1 || diff < -1)
			return false;

		return IsBalanced_Solution(root.left)
				&& IsBalanced_Solution(root.right);
	}

	private int TreeDepth(TreeNode root) {

		if (root == null) {
			return 0;
		}

		int leftDepth = TreeDepth(root.left);
		int rightDepth = TreeDepth(root.right);

		return leftDepth > rightDepth ? (leftDepth + 1) : (rightDepth + 1);
	}

方法2：

//后续遍历时，遍历到一个节点，其左右子树已经遍历  依次自底向上判断，每个节点只需要遍历一次

private boolean isBalanced = true;

	/**
	 * 后续遍历时，遍历到一个节点，其左右子树已经遍历 依次自底向上判断，每个节点只需要遍历一次
	 * 
	 * @param root
	 * @return
	 */
	public boolean IsBalanced_Solution1(TreeNode root) {

		getDepth(root);
		return isBalanced;
	}

	public int getDepth(TreeNode root) {
		if (root == null)
			return 0;
		int left = getDepth(root.left);
		int right = getDepth(root.right);

		if (Math.abs(left - right) > 1) {
			isBalanced = false;
		}
		return right > left ? right + 1 : left + 1;

	}