java-判断二叉树是不是平衡

参考了 http://zhedahht.blog.163.com/blog/static/25411174201142733927831/
但是用java来实现有一个问题。
由于Java无法像C那样“传递参数的地址,函数返回时能得到参数的值”,唯有新建一个辅助类:AuxClass

import ljn.help.*;
public class BalancedBTree {

	/**
	 * Q判断二叉树是不是平衡
 		1
	   / \
	  2   3
     / \   \
    4   5   6
       /
      7
	 */
	public static void main(String[] args) {

		int[][] threeBTrees={
				{1,2,3,4,5,0,6,0,0,7},//balanced
				{1,2,3,4,5,0,0,0,0,7},//not balanced
				};
		for(int[] each:threeBTrees){
			Node node=Helper.createTree(each);
			AuxClass aux= new AuxClass();
			boolean result=isBalanced(node,aux);
			System.out.println(result);
		}
	}

	/*
	 * 如果我们用后序遍历的方式遍历二叉树的每一个结点,在遍历到一个结点之前我们已经遍历了它的左右子树。
	 * 只要在遍历每个结点的时候记录它的深度(某一结点的深度等于它到叶节点的路径的长度),
	 * 我们就可以一边遍历一边判断每个结点是不是平衡的
	 * C/C++代码可参见 http://zhedahht.blog.163.com/blog/static/25411174201142733927831/
	 * 由于Java无法像C那样“传递参数的地址,函数返回时能得到参数的值”,唯有新建一个辅助类:AuxClass
	 */
	public static boolean isBalanced(Node node,AuxClass aux){
		if(node==null){
			aux.depth=0;
			return true;
		}
		AuxClass left=new AuxClass();
		AuxClass right=new AuxClass();
		//get leftTreeDepth and rightTreeDepth of a node.If the 'diff' is bigger than 1,return false;true otherwise
		if(isBalanced(node.getLeft(),left)&&isBalanced(node.getRight(),right)){
			int leftDepth=left.depth;
			int rightDepth=right.depth;
			int diff=leftDepth-rightDepth;
			if(diff==1||diff==-1||diff==0){
				aux.depth=leftDepth>rightDepth?leftDepth+1:rightDepth+1;
				return true;
			}
		}
		return false;
	}
	public static class AuxClass{
		private int depth;
	}
}


你可能感兴趣的:(java)