题目描述
思路分析
二叉树+递归
递归要返回一个
b
o
o
l
bool
bool和一个高度,把
b
o
o
l
bool
bool设置为全局变量即可。或者返回
p
a
i
r
pair
pair或者
v
e
c
t
o
r
vector
vector也行。
代码实现
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool res;
bool isBalanced(TreeNode* root) {
res=true;
dfs(root);
return res;
}
int dfs(TreeNode* root){
if(!root) return 0;
int lh=dfs(root->left),rh=dfs(root->right);
if(abs(lh-rh)>1) res=false;
return max(lh,rh)+1;
}
};