记录一下被自己蠢哭的代码(1)

题目如下

给定一个二叉树,判断它是否是高度平衡的二叉树。

本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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;
    }
}
posted @ 2024-02-17 21:06  CV小能手chh  阅读(4)  评论(0)    收藏  举报  来源