LeetCode 110 判断平衡二叉树

Leetcode 110 判断平衡二叉树

给定一颗二叉树，判断该树是否满足平衡条件: 对于任一节点，其左右子树高度相差不超过1

DFS

执行用时：1 ms, 在所有 Java 提交中击败了99.73%的用户
内存消耗：39.8 MB, 在所有 Java 提交中击败了70.51%的用户

class Solution {
    public boolean isBalanced(TreeNode root) {
        //空节点
        if(root==null) {
            return true;
        }
        //叶子节点
        else if(root!=null && root.left==null && root.right==null) {
            return true;
        }
        //非叶子节点
        else {
            return (Math.abs(dfs(root.left)-dfs(root.right))<2) && isBalanced(root.left) && isBalanced(root.right);
        }

    }

    //对于根节点root，返回其左右子树(如果有的话)的高度
    public int dfs(TreeNode root) {
        if(root==null) {
            return 0;
        }
        int leftHeight = dfs(root.left);
        int rightHeight = dfs(root.right);
        return Math.max(leftHeight, rightHeight)+1;
    }

}