题目描述
描述:实现一个函数,检查二叉树是否平衡。在这个问题中,平衡树的定义如下:任意一个节点,其两棵子树的高度差不超过 1。
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。
解题思路
思路1:最直观的想法是,编写一个dfs函数,实现后序遍历求当前节点高度。检查平衡性即当前节点的左右子树高度差绝对值小于等于1并且左子树平衡并且右子树平衡。由于该题没有告诉节点个数范围,故我没有使用记忆化搜索,其实dfs求高度可以使用记忆化搜索优化时间。
int dfs(TreeNode* cur) //后序遍历求高度
{
if(!cur) return 0; //空节点
int left=dfs(cur->left); //左
int right=dfs(cur->right); //右
return max(left,right)+1; //高度
}
bool isBalanced(TreeNode* root) {
if(!root) return true; //空树
return abs(dfs(root->left)-dfs(root->right))<=1&&isBalanced(root->left)&&isBalanced(root->right);
}
思路2:上述是先求高度然后再判断平衡,其实可以边求高度边判断平衡。平衡即左右子树各自平衡,且当前节点的左右子树高度差小于等于1,而后序遍历求高度本身即是后序遍历,其会先遍历左右子树,故可以在遍历左右子树的过程中,判断左右子树的左右子树高度差是否小于等于1,如果不是直接返回-1标记不平衡。
int dfs(TreeNode* cur) //后序遍历求高度
{
if(!cur) return 0; //空节点
int left=dfs(cur->left); //左
if(left==-1)
return -1;
int right=dfs(cur->right); //右
if(right==-1)
return -1;
int diff=abs(left-right);
if(diff>1) return -1;
return max(left,right)+1; //高度
}
bool isBalanced(TreeNode* root) {
if(!root) return true; //空树
return dfs(root)==-1?false:true;
}
总结:先完成,再优化。