题目如下
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4] 输出:false
示例 3:
输入:root = [] 输出:true
提示:
- 树中的节点数在范围
[0, 5000]
内 -104 <= Node.val <= 104
思路,深度遍历确定所有节点为根节点的子树符合要求
朴素(若至)的代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isBalanced(TreeNode root) {
if(root!=null){
boolean is= isBalanced(root.left)&isBalanced(root.right);
if(Math.abs(deeps(root.left)-deeps(root.right))<=1)return true&is;
return false;
}
return true;
}
int deeps(TreeNode root){
if(root==null)return 0;
int i=deeps(root.left),j=deeps(root.right);
return i>j?i+1:j+1;
}
}
真就被自己蠢哭了,每次判断需要进行一次探索左右子树深度,O(n2)恐怖的达到了126ms
大佬代码
/*递归方法*/
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
return helper(root) != -1;
}
private int helper(TreeNode root){
if(root == null){
return 0;
}
int left = helper(root.left);
int right = helper(root.right);
if (left == -1 || right == -1 || Math.abs(left-right) > 1){
return -1;
}
return Math.max(left,right)+1;
}
}