leetcode-110-Balanced Binary Tree

问题

题目：[leetcode-110]

思路1

这个题其实思路也很清晰，就是枚举每一个点即可。
所以本质是遍历。只不过遍历的时候，每个点需要判断是不是平衡的。这个操作我单独写visit函数即可。
上次的path sum iii对我在遍历有了更进一步的认识。

代码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;

        bool ret = visit(root);
        if(!ret) return false;
        ret = isBalanced(root->left);
        if(!ret) return false;
        ret = isBalanced(root->right);
        return ret;

    }
private:
    bool visit(TreeNode* root){
        if(!root) return true;
        int left = depth(root->left);
        int right = depth(root->right);
        return abs(left - right) < 2;
    }
    int depth(TreeNode* root){
        if(!root) return 0;
        return std::max(depth(root->left), depth(root->right)) + 1;
    }
};

---

思路2

参考了别人的办法，发现这个题O(N)的办法是把语义换了。
判断一棵树是否是平衡的，只需判断：

• 左子树是否平衡
• 右子树是否平衡
• 当前根节点是否平衡

只不过这个函数写的时候，如果平衡，返回的是树的实际高度。

代码2

/**
 * 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) {
        return -1==checkDepth(root)?false:true;
    }
private:
    int checkDepth(TreeNode* root){ // 判断子树是否平衡，如果平衡，返回高度。否则，返回-1
        if(!root) return 0;
        int left = checkDepth(root->left);
        if(-1==left) return -1;
        int right = checkDepth(root->right);
        if(-1==right) return -1;

        int diff = abs(left - right);
        if(diff>1) return-1;

        return std::max(left, right) + 1;
    }
};