剑指offer | 面试题42:平衡二叉树

2022-02-23 15:38:10 浏览数 (1)

死磕算法系列文章

  1. 干货 | 手撕十大经典排序算法
  2. 剑指offer | 认识面试
  3. 剑指offer | 面试题2:实现Singleton模式
  4. 剑指offer | 面试题3:二维数组的查找
  5. 剑指offer | 面试题4:替换空格
  6. 剑指offer | 面试题5:从尾到头打印链表
  7. 剑指offer | 面试题6:重建二叉树
  8. 剑指offer | 面试题7:用两个栈实现队列
  9. 剑指offer | 面试题8:旋转数组的最小数字
  10. 剑指offer | 面试题9:斐波那契数列
  11. 剑指offer | 面试题10:青蛙跳台阶问题
  12. 剑指offer | 面试题11:矩阵覆盖
  13. 剑指offer | 面试题12:二进制中1的个数
  14. 剑指offer | 面试题13:数值的整数次方
  15. 剑指offer | 面试题14:打印从1到最大的n位数
  16. 剑指offer | 面试题15:删除链表的节点
  17. 剑指offer | 面试题16:将数组中的奇数放在偶数前
  18. 剑指offer | 面试题17:链表中倒数第k个节点
  19. 剑指offer | 面试题18:反转链表
  20. 剑指offer | 面试题19:合并两个有序链表
  21. 剑指offer | 面试题20:判断二叉树A中是否包含子树B
  22. 剑指offer | 面试题21:二叉树的镜像
  23. 剑指offer | 面试题22:顺时针打印矩阵
  24. 剑指offer | 面试题23:包含min函数的栈
  25. 剑指offer | 面试题24:栈的压入、弹出序列
  26. 剑指offer | 面试题25:从上到下打印二叉树
  27. 剑指offer | 面试题26:二叉搜索树的后序遍历序列
  28. 剑指offer | 面试题27:二叉树中和为某一值的路径
  29. 剑指offer | 面试题28:复杂链表的复制
  30. 剑指offer | 面试题29:二叉搜索树转换为双向链表
  31. 剑指offer | 面试题30:字符串的排列
  32. 剑指offer | 面试题31:数组中出现次数超过一半的数字
  33. 剑指offer | 面试题32:最小的k个数
  34. 剑指offer | 面试题33:连续子数组的最大和
  35. 剑指offer | 面试题34:1~n 整数中 1 出现的次数
  36. 剑指offer | 面试题35:把数组排成最小的数
  37. 剑指offer | 面试题36:丑数
  38. 剑指offer | 面试题37:第一个只出现一次的字符
  39. 剑指offer | 面试题38:数组中的逆序对
  40. 剑指offer | 面试题39:两个链表的第一个公共节点
  41. 剑指offer | 面试题40:数组中数字出现的次数
  42. 剑指offer | 面试题41:二叉树的深度

Leetcode : https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof/

GitHub : https://github.com/nateshao/leetcode/blob/main/algo-notes/src/main/java/com/nateshao/sword_offer/topic_42_isBalanced/Solution.java

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

题目描述 :输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。 难度:简单

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

代码语言:javascript复制
例如:给定二叉树 [3,9,20,null,null,15,7],

    3
   / 
  9  20
    /  
   15   7
返回 true 。

示例 2:

代码语言:javascript复制
给定二叉树 [1,2,2,3,3,null,null,4,4]

           1
          / 
         2   2
        / 
       3   3
      / 
     4   4
    返回 false 。

此题为 剑指offer | 面试题41:二叉树的深度 的拓展

方法一:先序遍历 判断深度 (从顶至底)

思路:平衡二叉树的条件:左子树是平衡二叉树,右子树是平衡二叉树,左右子树高度不超过 1。

算法流程:

isBalanced(root) 函数: 判断树 root 是否平衡

  • 特例处理: 若树根节点 root 为空,则直接返回 true ;

返回值: 所有子树都需要满足平衡树性质,因此以下三者使用与逻辑 && 连接;

  1. Math.abs(depth(root.left) - depth(root.right)) <= 1 :判断 当前子树 是否是平衡树;
  2. isBalanced(root.left) :先序遍历递归,判断 当前子树的左子树 是否是平衡树;
  3. isBalanced(root.right) :先序遍历递归,判断 当前子树的右子树 是否是平衡树;

depth(TreeNode root) 函数: 计算树 root 的深度

  • 终止条件:root 为空,即越过叶子节点,则返回高度 0 ;
  • 返回值: 返回左 / 右子树的深度的最大值 1 。
复杂度分析:
  • **时间复杂度 O(N log N):**最差情况下(为 “满二叉树” 时), isBalanced(root) 遍历树所有节点,判断每个节点的深度 depth(root)需要遍历 各子树的所有节点 。
  • 空间复杂度 O(N): 最差情况下(树退化为链表时),系统递归需要使用 O(N) 的栈空间。

Java

代码语言:javascript复制
package com.nateshao.sword_offer.topic_42_isBalanced;

/**
 * @date Created by 邵桐杰 on 2022/1/23 19:26
 * @微信公众号 千羽的编程时光
 * @个人网站 www.nateshao.cn
 * @博客 https://nateshao.gitee.io
 * @GitHub https://github.com/nateshao
 * @Gitee https://gitee.com/nateshao
 * Description:
 */
public class Solution {

    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
            return Math.abs(depth(root.left) - depth(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
    }

    private int depth(TreeNode root) {
        if (root == null) return 0;
        return Math.max(depth(root.left), depth(root.right))   1;
    }

//    public boolean isBalanced(TreeNode root){
//        if(root == null) {
//            return true;
//        }
//        boolean condition = Math.abs(maxDepth(root.left) - maxDepth(root.right)) <= 1;
//        return condition && isBalanced(root.left) && isBalanced(root.right);
//    }
//    public int maxDepth(TreeNode root) {
//        if (root == null) {
//            return 0;
//        }
//        int left = maxDepth(root.left);
//        int right = maxDepth(root.right);
//        return Math.max(left, right)   1;
//    }
    public class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
 }
}

Go

代码语言:javascript复制
func isBalanced(node *TreeNode) bool {
 return node == nil || isBalanced(node.Left) &&
 math.Abs(height(node.Left)-height(node.Right)) <= 1  && // 左右子树高度不超过 1
  isBalanced(node.Right)
}

//计算节点最大深度
func height(node *TreeNode) float64 {
 if node == nil {
  return 0
 }
 return math.Max(height(node.Left), height(node.Right))   1
}

方法二:后序遍历 剪枝 (从底至顶)

“思路是对二叉树做后序遍历,从底至顶返回子树深度,若判定某子树不是平衡树则 “剪枝” ,直接向上返回。

算法流程:

recur(root) 函数:

  • 返回值:

当节点root 左 / 右子树的深度差 ≤1 :则返回当前子树的深度,即节点 root 的左 / 右子树的深度最大值 1 ( max(left, right) 1 );

  1. 当节点root 左 / 右子树的深度差 >2 :则返回 −1 ,代表 此子树不是平衡树

终止条件:

  1. root 为空:说明越过叶节点,因此返回高度 0 ;

当左(右)子树深度为 -1−1 :代表此树的 左(右)子树 不是平衡树,因此剪枝,直接返回 −1 ;isBalanced(root) 函数:

  • 返回值:recur(root) != -1 ,则说明此树平衡,返回 true ;否则返回 false 。

复杂度分析:

  • 时间复杂度 O(N): N为树的节点数;最差情况下,需要递归遍历树的所有节点。
  • 空间复杂度 O(N): 最差情况下(树退化为链表时),系统递归需要使用 O(N) 的栈空间。
代码语言:javascript复制
class Solution {
    public boolean isBalanced(TreeNode root) {
        return recur(root) != -1;
    }

    private int recur(TreeNode root) {
        if (root == null) return 0;
        int left = recur(root.left);
        if(left == -1) return -1;
        int right = recur(root.right);
        if(right == -1) return -1;
        return Math.abs(left - right) < 2 ? Math.max(left, right)   1 : -1;
    }
}

参考链接:https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof/solution/mian-shi-ti-55-ii-ping-heng-er-cha-shu-cong-di-zhi/

0 人点赞