二叉树 | 判断是否为平衡二叉树

牛客BM 36 判断是否为平衡二叉树

题目:给你二叉树的根节点root,判断这个树是否为平衡二叉树。

定义:二叉树的左右子树的高度之差的绝对值不超过1;平衡二叉树的子树也是平衡二叉树。

思路:递归
遍历方式:后序遍历
1.分别求左右子树的高度;
2. 若左右子树高度差不符合定义,则返回-1;
3.若左右子树高度差符合定义,则返回左右子树高度的最大值+1,作为当前结点的高度,递归向上判断;

举例
1.以2为根结点的左右子树,均符合平衡二叉树的定义,则以2为根结点的二叉树为平衡二叉树;
2.以3为根结点的二叉树同样符合平衡二叉树的定义
3.递归向上判断:则以1为根结点的二叉树符合平衡二叉树的定义
二叉树 | 判断是否为平衡二叉树_第1张图片

代码

	/*判断是否为平衡二叉树*/
    public boolean IsBalanced_Solution(TreeNode root) {
    	if(root == null) return true;//空树
    	return (getDepth(root) == -1) ? false : true;//返回-1就是错的
    }
    
    /*获取子树高度,平衡则返回正常的高度,不平衡返回-1*/
    public static int getDepth(TreeNode root) {
    	if(root == null) return 0;//当前结点为空
    	//左
    	int left = getDepth(root.left);//合法则返回正常高度,非法返回-1
    	if(left == -1) return -1;//左子树非法
    	//右
    	int right = getDepth(root.right);//合法则返回正常高度,非法返回-1
    	if(right == -1) return -1;//右子树非法
    	//根
    	if(Math.abs(left - right) > 1) return -1;//根非法
    	return 1 + Math.max(left, right);//根结点的高度
    }//getDepth

你可能感兴趣的:(算法,Coding,算法,数据结构,java)