代码随想录之平衡二叉树

代码随想录之平衡二叉树_第1张图片
本题思路是针对高度,只不过是判断高度之差的绝对值是否大于1罢了,这里引入特殊变量-1用来说明不是平衡二叉树
如果左右子节点对应的子树有一个不是平衡二叉树就一直-1返回到头了。
只有全是平衡二叉树时才正常运行完。

class Solution {
public:
    int traversal(TreeNode* cur){
        if(cur == nullptr) return 0;
        int l_dept = traversal(cur->left);
        if(l_dept == -1) return -1;
        int r_dept = traversal(cur->right);
        if(r_dept == -1) return -1;

        return abs(r_dept-l_dept)>1? -1:1+max(r_dept,l_dept);        
    }
    bool isBalanced(TreeNode* root) {
        return traversal(root) != -1;
    }
};

你可能感兴趣的:(代码之二叉树,c++,算法,数据结构)