判断是否是平衡二叉树

输入一棵二叉树,判断该二叉树是否是平衡二叉树。

我本来想的是用dfs,每个节点递归,然后回溯看右节点,但是发现不会写。这个递归还是没有理解好,dfs跟bfs也不熟练,这两个方法使用来遍历图的,图是树的特殊状态(个人观点),所以主要还是算法的本质掌握的不熟练。那这道题呢,有两个方法。

1.平衡二叉树是任意节点的左右子树深度相差不能为1的,所以我们可以计算每个节点的深度,来看这颗树是不是平衡的。

class Solution
{
public:
    bool IsBalanced_Solution(TreeNode* pRoot)
    {
        if(!pRoot) return true;
        int left = getdepth(pRoot->left);
        int right = getdepth(pRoot->right);
        if(abs(left-right)>1)
            return false;
        return(IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right));
    }

    //递归实现,dfs
    int getdepth(TreeNode* pRoot)
    {
        if(!pRoot) return 0;
        int left = getdepth(pRoot->left);
        int right = getdepth(pRoot->right);
        return max(left+1,right+1);
    }

    //迭代,bfs
    int getdepth(TreeNode* pRoot)
    {
        if(!pRoot)  return 0;
        queue<TreeNode*> que;
        que.push(pRoot);
        int depth =0;
        while(!que.empty())
        {
            int size = que.size();
            ++depth;
            for(int i=0;i<que.size();++i)
            {
                TreeNode* temp = que.front();
                que.pop();
                if(temp->left)
                    que.push(temp->left);
                if(temp->right)
                    que.push(temp->right);
            }
        }
        return depth;
    }
}

那么这个呢是根据每个节点的深度来判断这棵树是不是平衡二插树。算深度的算法使用了迭代和递归两种方法来实现。这个算法有个问题就是,判断true/false只在is那个函数里面判断了,所以我们要每个节点都遍历,这样下层的节点实际上我们就走了很多次。这个是很浪费时间的。所以就有第二个算法。

2

class Solution
{
public:
    bool status = true;

    bool IsBalanced_Solution(TreeNode* pRoot)
    {
        getdepth(pRoot);
        return status;
    }

    int getdepth(TreeNode* pRoot)
    {
        if(!pRoot) return 0;
        int left=getdepth(pRoot->left);
        int right=getdepth(pRoot->right);
        if(abs(left-right)>1)
            status = false;
        return max(left+1,right+1);
    }

}

这个方法看上去跟上一个类似,但是还有区别的。上一个只在is函数里面才有false/true,并且每个is函数都要调用一下getdepth函数。所以每个节点会计算很多啊遍。
而第二个方法在getdepth中有true和false,只要有一个节点不满足平衡就返回false,并且getdepth只计算了一次全部的节点,没有多余的计算。

以上。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 树的概述 树是一种非常常用的数据结构,树与前面介绍的线性表,栈,队列等线性结构不同,树是一种非线性结构 1.树的定...
    Jack921阅读 4,489评论 1 31
  • 1 序 2016年6月25日夜,帝都,天下着大雨,拖着行李箱和同学在校门口照了最后一张合照,搬离寝室打车去了提前租...
    RichardJieChen阅读 5,165评论 0 12
  • 一直以来,我都很少使用也避免使用到树和图,总觉得它们神秘而又复杂,但是树在一些运算和查找中也不可避免的要使用到,那...
    24K男阅读 6,782评论 5 14
  • 二叉树 在计算机科学中,二叉树是每个节点最多有两个子树的树结构。通常子树被称作“左子树”和“右子树”,左子树和右子...
    静默加载阅读 2,059评论 0 3
  • 基于树实现的数据结构,具有两个核心特征: 逻辑结构:数据元素之间具有层次关系; 数据运算:操作方法具有Log级的平...
    yhthu阅读 4,320评论 1 5