判断一个二叉树是否为平衡二叉树(树形dp)

平衡二叉树的严格定义是,一棵二叉树的任何一个结点,左子树和右子树的高度差不超过一,即为平衡二叉树。
判断是否平衡二叉树的大体思路是,以每一个结点为头结点的二叉树都是平衡二叉树,那整棵树都是平衡二叉树,

(1)判断左树是否平衡,如果不平衡返回false
(2)判断右树是否平衡,如果不平衡返回false
(3)拿到左树的高度
(4)拿到右树的高度
(5)判断左右树高度差是否超过一,超过一,返回false
下面就可以根据上述的这些信息,设计递归返回结构,遍历左树的过程,应该给我返回左树是否平衡,以及左树高度这两个信息。右树需要返回的信息也一样,所以整理出递归函数的返回值,应该包括高度,和是否平衡这两个信息

下面看代码:

class isBalancedTree{
private:
    class Node{
    public:
        int value;
        Node *left;
        Node *right;
        Node(int data){
            this->value=data;
        }
    };
    class returnData{
    public:
        bool isBalance;
        int h;
        returnData(bool isB,int h){
            this->isBalance=isB;
            this->h=h;
        }
    };
     returnData process(Node *head){
          if(head== nullptr)
              return returnData(true,0);

          returnData leftData =process(head->left);
          if(!leftData.isBalance)
              return returnData(false,0);

          returnData rightData=process(head->right);
          if(!leftData.isBalance)
              return returnData(false,0);

          if(abs(leftData.h -  rightData.h)>1)
              return returnData(false,0);
          return returnData(true,max(leftData.h,rightData.h)+1);
     }
     bool isB(Node *head){
         return process(head).isBalance;
     }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值