LintCode 平衡二叉树

题目描述:
给定一个二叉树,确定它是高度平衡的。对于这个问题,一棵高度平衡的二叉树的定义是:一棵二叉树中每个节点的两个子树的深度相差不会超过1。

您在真实的面试中是否遇到过这个题? Yes
样例
给出二叉树 A={3,9,20,#,#,15,7}, B={3,#,20,15,7}

二叉树A是高度平衡的二叉树,但是B不是。

思路分析:

遍历每个结点的左右子树。判断深度差的绝对值。

ac代码:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of binary tree.
     * @return: True if this Binary tree is Balanced, or false.
     */
    int depth(TreeNode *root)
    {
        if (root == NULL)
            return 0;
        return max(depth(root->left), depth(root->right)) + 1;
    }

    bool isBalanced(TreeNode *root) {
        // write your code here
        if (root == NULL)
            return true;
        if (!isBalanced(root->left))
            return false;
        if (!isBalanced(root->right))
            return false;
        int left = depth(root->left);
        int right = depth(root->right);
        if (abs(left-right) > 1)
            return false;

        return true;

    }
};
发布了166 篇原创文章 · 获赞 4 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/sinat_34336698/article/details/70038026