56. 2种方法判断二叉树是不是平衡二叉树[is balanced tree]

最新推荐文章于 2023-01-04 18:48:52 发布

技术搬砖工

最新推荐文章于 2023-01-04 18:48:52 发布

阅读量913

点赞数

分类专栏： c++ 文章标签： algorithm 二叉树 c++

c++ 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了如何通过两种方法来判断给定二叉树是否为平衡二叉树。第一种方法的时间复杂度为O(n^2)，通过递归判断每个节点的左右子树是否平衡并求取深度；第二种方法改进了时间复杂度至O(n)，在判断平衡的同时计算深度。文章还提供了C++代码实现。

摘要由CSDN通过智能技术生成

【本文链接】

http://www.cnblogs.com/hellogiser/p/is-balanced-tree.html

【题目】

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。例如下图中的二叉树就是一棵平衡二叉树：

【分析】

之前的博文27.二元树的深度[BinaryTreeDepth]中介绍过如何求二叉树的深度。有了经验之后再解决这个问题，我们很容易就能想到思路。

【方案1】

先判断左右子树是不是平衡的，若平衡再求出左右子树的深度，若深度之差大于1，则不平衡。因为在遍历每个结点时都要求其左右子树的深度，因此时间复杂度是O(n^2)的。

【代码】

      C++ Code  
   

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

#include "stdafx.h"
#include <cmath>
#include <algorithm>

/*
    version: 1.0
    author: hellogiser
    blog: http://www.cnblogs.com/hellogiser
    date: 2014/5/25
*/

// binary tree node struct
struct BinaryTreeNode
{
     int value;
    BinaryTreeNode *left;
    BinaryTreeNode *right;
};

// Get depth of a binary tree
int TreeDepth(BinaryTreeNode *root)
{
     // the depth of a empty tree is 0
     if( NULL == root)
         return 0;

     // the depth of left sub-tree
     int nLeft = TreeDepth(root->left);
     // the depth of right sub-tree
     int nRight = TreeDepth(root->right);

     // depth is the binary tree
     return (nLeft > nRight) ? (nLeft + 1) : (nRight + 1);
     // return max(nLeft,nRight)+1;
}

// is balanced tree in O(n^2)
bool IsBalanced(BinaryTreeNode *root)
{
     if( NULL == root)
         return true;
     if(!IsBalanced(root->left))
         return false;
     if(!IsBalanced(root->right))
         return false;
     int leftDepth = TreeDepth(root->left);
     int rightDepth = TreeDepth(root->right);
     if (abs(leftDepth - rightDepth) > 1)
         return false;
     else
         return true;
}

【方案2】

在判断左右子树是否平衡的过程中把深度计算出来，这样在对父结点进行平衡判断时就可以不用再重复计算左右子树的深度了。其时间复杂度为O(n)。

【代码】

      C++ Code  
   

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

// is balanced tree in O(n)
bool IsBalanced(BinaryTreeNode *root, int &depth)
{
     if( NULL == root)
    {
        depth = 0;
         return true;
    }

     int leftDepth, rightDepth;
     if(!IsBalanced(root->left, leftDepth))
         return false;
     if(!IsBalanced(root->right, rightDepth))
         return false;

     // get root depth without visiting left and right sub-trees
    depth = (leftDepth > rightDepth) ? (leftDepth + 1) : (rightDepth + 1);
     if (abs(leftDepth - rightDepth) > 1)
         return false;
     else
         return true;
}

// is balanced tree
bool IsBalancedTree(BinaryTreeNode *root)
{
     int depth;
     return IsBalanced(root, depth);
}

【参考】

http://zhedahht.blog.163.com/blog/static/25411174201142733927831/

http://blog.csdn.net/zjull/article/details/11646591

http://blog.csdn.net/luckyxiaoqiang/article/details/7518888

【本文链接】

http://www.cnblogs.com/hellogiser/p/is-balanced-tree.html