剑指 Offer 55 - II. 平衡二叉树

题目

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。
限制：
0 <= 树的结点个数 <= 10000

思路

DFS经典应用

运行结果

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    boolean flag  = true;
	public boolean isBalanced(TreeNode root) {
		f(root);
		return flag;
    }
	public int f(TreeNode root) {
		if(root == null) return 0;
		int left = f(root.left);
		int right = f(root.right);
		if(left-right<-1||left-right>1) {
			flag = false;
		}
		
		return (left>right?left:right)+1;
	}
}