LeetCode110. 平衡二叉树

快来打我* 2022-05-09 11:04 180阅读 0赞

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

本题中，一棵高度平衡二叉树定义为：

> 一个二叉树*每个节点* 的左右两个子树的高度差的绝对值不超过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` 。

--------------------

思路：遍历二叉树，判断二叉树每个结点的左右子树的高度差是否小于1。

/**
     * Definition for a binary tree node.
     * public class TreeNode {
     *     int val;
     *     TreeNode left;
     *     TreeNode right;
     *     TreeNode(int x) { val = x; }
     * }
     */
    class Solution {
        public boolean isBalanced(TreeNode root) {
              if(null==root){
                return true;
            }
            ArrayDeque<TreeNode> queue=new ArrayDeque<TreeNode>();
            queue.add(root);
            while(!queue.isEmpty()){
                TreeNode node=queue.removeFirst();
                int leftChirdHeight=getTreeHeight(node.left);
                int rightChirdHeight=getTreeHeight(node.right);
                if(Math.abs(leftChirdHeight-rightChirdHeight)>1){
                    return false;
                }
                if(null!=node.left){
                    queue.add(node.left);
                }
                if(null!=node.right){
                    queue.add(node.right);
                }
            }
            return true;
        }
        /**
         * 非递归求二叉树深度
         * @param root
         * @return
         */
         public  int getTreeHeight(TreeNode root){
            if(null==root){
                return 0;
            }
            ArrayDeque<TreeNode> queue=new ArrayDeque<TreeNode>();
            int height=0;
            queue.add(root);
            while(!queue.isEmpty()){
                int size=queue.size();
                for(int i=0;i<size;i++){
                    TreeNode node=queue.removeFirst();
                    if(null!=node.left){
                        queue.add(node.left);
                    }
                    if(null!=node.right){
                        queue.add(node.right);
                    }
                }
                height++;
            }
            return height;
        }
    }