判断二叉树是否平衡,平衡并返回高度

本文介绍了一种用于判断二叉树是否为平衡二叉树的递归算法,并通过具体的示例展示了该算法的应用过程。平衡二叉树是指任意节点的左右子树高度差不超过1的二叉树。
public class IsBalanceTree {
	static class Node{
		int value;
		Node left;
		Node right;
		public Node(int data) {
			value = data;
		}
	}
	static class ReturnData{
		boolean isB;
		int level;
		public ReturnData(boolean isB, int level) {
			this.isB = isB;
			this.level = level;
		}
	}
	public static ReturnData isBalance(Node head) {
		if(head == null) {
			return new ReturnData(true,0);
		}
		ReturnData left = isBalance(head.left);
		if(!left.isB) {
			return new ReturnData(false,0);
		}
		ReturnData right = isBalance(head.right);
		if(!right.isB) {
			return new ReturnData(false,0);
		}
		if(Math.abs(left.level - right.level) > 1) {
			return new ReturnData(false,0);
		}
		return new ReturnData(true, Math.max(left.level, right.level) + 1);
	}
	public static void printIsBalance(ReturnData rd) {
		if(rd.isB) {
			System.out.println("true," + "level:" + rd.level);
		}else {
			System.out.println("false");
		}
	}
	
	//for test
	public static void main(String[] args) {
		Node head = null;
		printIsBalance(isBalance(head));;
		
		head = new Node(1);
		printIsBalance(isBalance(head));;
		
		head = new Node(1);
		head.left = new Node(2);
		head.right = new Node(3);
		head.left.left = new Node(4);
		head.left.left.left = new Node(5);
		printIsBalance(isBalance(head));;
		
		head = new Node(1);
		head.left = new Node(2);
		head.right = new Node(3);
		head.left.left = new Node(4);
		head.left.right = new Node(6);
		printIsBalance(isBalance(head));;
		
		
		head = new Node(100);
		head.left = new Node(21);
		head.left.left = new Node(37);
		head.right = new Node(-42);
		head.right.left = new Node(0);
		head.right.right = new Node(666);
		head.right.right.right = new Node(343432);
		head.right.right.right.right = new Node(4343434);
		printIsBalance(isBalance(head));;
	}
	
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值