【LeetCode】平衡二叉树

题目描述:

https://leetcode-cn.com/problems/balanced-binary-tree/

解题思路:

递归;
第一步:找终止条件,即root==NULL,返回true;
第二步:找返回值,即应该给上级返回什么信息;返回该节点作为根结点的数是否为平衡树
第三步:本级需要做什么:首先,需要判断左右子树的高度差是否大于1,大于着返回false;再次,判断左子树和右子树是否都是平衡术,若一个不是,返回false;否则返回true;
其中需要计算某个节点所在层树,又要利用递归,即若root==NULL,返回0;否则,返回其左右节点层树的大值加1;

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(!root)
            return true;
        if(abs(countFloor(root->left) - countFloor(root->right)) > 1)
            return false;
        else if(!isBalanced(root->left) || !isBalanced(root->right))
            return false;
        else return true;
        
    }
    int countFloor(TreeNode* root)
    {
        if(!root)
            return 0;
        return max(countFloor(root->left), countFloor(root->right)) + 1;
    }
};

补充:递归思想参考http://39.96.217.32/blog/4,我觉得讲的挺易懂的;

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