剑指 offer第55.2题-平衡二叉树

在这里插入图片描述

平衡二叉树左右子树高度差不超过 1。

分别计算左右子树的高度,若是差值不超过1,则为平衡二叉树。这是最直接的做法,遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。

public classSolution {
    public boolean IsBalanced_Solution(TreeNode root) {
        if(root == null) {
            return true;
        }
        return Math.abs(maxDepth(root.left) - maxDepth(root.right)) <= 1 &&
            IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
    }
      
    private int maxDepth(TreeNode root) {
        if(root == null) {
            return 0;
        }
        return 1 + Math.max(maxDepth(root.left), maxDepth(root.right));
    }
}

这种做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。

用后序遍历的方法遍历二叉树,则是先遍历了左、右子树,遍历每个节点的时候记录深度,就可以判断每个节点是不是平衡的。

public class Solution {
    private boolean isBalanced=true;
    public boolean IsBalanced_Solution(TreeNode root) {
        height(root);
        return isBalanced;
    }
    private int height(TreeNode root){
        if(root==null||!isBalanced){
            return 0;
        }
        int left = height(root.left);
        int right=height(root.right);
        if(Math.abs(left-right)>1)
           isBalanced=false;
        return 1+Math.max(left,right);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是用C语言实现的代码: ```c #include <stdio.h> #include <stdlib.h> // 二叉树结点结构体 typedef struct TreeNode { int val; // 结点值 struct TreeNode* left; // 左子结点 struct TreeNode* right; // 右子结点 } TreeNode; // 插入结点 TreeNode* insert(TreeNode* root, int val) { if (root == NULL) { // 如果空树,则插入到根结点 TreeNode* node = (TreeNode*)malloc(sizeof(TreeNode)); node->val = val; node->left = NULL; node->right = NULL; return node; } if (val < root->val) { // 如果插入的值小于当前结点的值,则插入到左子树 root->left = insert(root->left, val); } else { // 否则插入到右子树 root->right = insert(root->right, val); } return root; } // 计算树的高度 int height(TreeNode* root) { if (root == NULL) return 0; int leftHeight = height(root->left); int rightHeight = height(root->right); return (leftHeight > rightHeight ? leftHeight : rightHeight) + 1; } // 打印二叉树 void printTree(TreeNode* root, int layer) { if (root == NULL) return; printTree(root->right, layer + 1); // 先打印右子树 for (int i = 0; i < layer - 1; i++) { printf(" "); } if (layer > 0) { printf(".--"); } printf("%d\n", root->val); printTree(root->left, layer + 1); // 再打印左子树 } int main() { TreeNode* root = NULL; // 根结点 int val; while (scanf("%d", &val) != EOF) { root = insert(root, val); // 插入结点 } int h = height(root); // 计算树的高度 printTree(root, h); // 打印二叉树 return 0; } ``` 输入样例: ``` 10 5 20 8 4 7 ``` 输出样例: ``` .---20 .--10 | `--8 | |--7 `--5 `--4 ``` 解释:该二叉树的结构如下: ``` 10 / \ 5 20 / \ 4 8 / 7 ``` 其中,叶子结点 4 和 7 的高度为 1,结点 5 和 8 的高度为 2,结点 10 的高度为 3,结点 20 的高度为 1。因此,树的高度为 3。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值