算法学习---平衡二叉树
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
leetCode题目链接:https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof/
看到题目,第一时间看到 二叉树的深度 定义为 从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
具体可以去看 二叉树的深度:https://leetcode-cn.com/problems/er-cha-shu-de-shen-du-lcof/
当学会了如何求解二叉树的深度时,那么只需要对比当前节点的左右子树的深度是否小于1
若小于1则继续向下遍历,若大于1了就说明这个二叉树已经不平衡了,可以返回false。
/**
* 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:
//求以该节点为根的二叉树的深度
int maxDepth(TreeNode* root) {
if (root == nullptr) return 0;
return max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
//遍历二叉树
bool isBalanced(TreeNode* root) {
if (root == nullptr) return true;
if (abs(maxDepth(root->left) - maxDepth(root->right)) <= 1)
return isBalanced(root->left) & isBalanced(root->right);
else
return false;
}
};