[C++]LeetCode: 35 Balanced Binary Tree

题目:

Given a binary tree, determine if it is height-balanced.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1

思路:递归判断每个结点。

Attention: 

1. 一旦发现不平衡,递归切断(前后切断);

2. 理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即使用不同的函数实现。

[C++]LeetCode: 35 Balanced Binary Tree_第1张图片

AC Code:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode *root) {
        //判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1
        //思路:递归判断每个结点。
        //Attention: 一旦发现不平衡,递归切断(前后切断);理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即时不同的函数实现。
        bool result = true;
        isBalanced_helper(root, result);
        return result;
    }

private:
    int isBalanced_helper(TreeNode* root, bool& result){
        if(root == NULL) return 0;
        
        //切断递归,一旦发现树不平衡,结束递归。
        if(result == false) return 0;
        
        int leftheight = isBalanced_helper(root->left, result);
        int rightheight = isBalanced_helper(root->right, result);
        
        // prevent overwrite of result 同时也是short-circuit
        if (result == false) return 0;
        
        if(abs(leftheight - rightheight) > 1) result = false;
        
        //计算当前节点的子树高度。
        return max(leftheight, rightheight) + 1;
        
    }
};



你可能感兴趣的:(LeetCode,tree,search,Depth-first)