判断平衡二叉树

package com.jsp.tree;
/**
 * 判断是否是平衡二叉树
 * @author jiangshipan
 * 任何一个节点 左子树和右子数高度差小于1 
 *  以每一个节点为头的树都是平衡的，则这个树平衡
 *  
 *  可能： 1.左树不平
 *  	  2.右树不平
 *  	  3. 左或右高  整棵树
 */
public class IsBalancedTree {
	public static class Node {
		public int value;
		public Node left;
		public Node right;
		
		public Node(int data) {
			this.value = data;
		}
	}
	
	public static class ReturnData {
		//返回两个信息 是否平衡 和 高度
		public boolean isB;
		public int h;
		
		public ReturnData(boolean isB, int h) {
			this.isB = isB;
			this.h = h;
		}
	}
	
	public static ReturnData process(Node head) {
		 if (head == null) {
			//空树是平衡的
			return new ReturnData(true, 0);
		 }
		 ReturnData leftData = process(head.left);
		 //左树不平衡了
		 if(!leftData.isB) {
			 //不需要高度了 所以设置成0
			 return new ReturnData(false, 0);
		 }
		 
		 ReturnData rightData = process(head.right);
		 //右树不平衡了
		 if(!rightData.isB) {
			 return new ReturnData(false, 0);
		 }
		 //左右树都平衡
		 if(Math.abs(leftData.h - rightData.h) > 1) {
			 return new ReturnData(false, 0);
		 } 
		 //高度为左右树较大的 + 1
		 return new ReturnData(true, Math.max(leftData.h, rightData.h )+ 1);
		
	}
	
	public static boolean isB(Node head) {
		return process(head).isB;
	}
	
}