剑指 Offer 55 - I. 二叉树的深度
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
提示:
节点总数 <= 10000
/**
* 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;
queue<TreeNode*> q;
int ans = 0;
q.push(root);
while(q.empty() == false){
ans ++;
for(int i = q.size() - 1; i >= 0; -- i){
TreeNode *cur = q.front();
q.pop();
if(cur->left != nullptr)q.push(cur->left);
if(cur->right != nullptr)q.push(cur->right);
}
}
return ans;
}
};
算法思路
经典的bfs
(层次遍历),用一个队列存储节点,然后逐层遍历即可得到结果。
剑指 Offer 55 - II. 平衡二叉树
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true
。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false
。
限制:
0 <= 树的结点个数 <= 10000
/**
* 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 ans = true;
int check(TreeNode *root){
if(root == nullptr)return 0;
int l = check(root->left), r = check(root->right);
ans = ans && (abs(l - r) <= 1);
return max(l, r) + 1;
}
bool isBalanced(TreeNode* root) {
if(root == nullptr)return true;
ans = true;
return abs(check(root->left) - check(root->right)) <= 1 && ans;
}
};
算法思路
平衡二叉树有两个条件:
- 左右子树都是平衡二叉树
- 左右子树的高度差的绝对值小于1
因此本题用递归可以很快解决,递归函数的返回值为树的深度,全局变量ans
则表示子树都是平衡二叉树,如果某个子树不是平衡二叉树,ans
将会更新为false
。
【剑指Offer】系列:
【剑指Offer】栈
【剑指Offer】链表
【剑指Offer】字符串
【剑指Offer】查找算法
【剑指Offer】查找算法(1)
【剑指Offer】搜索与回溯算法
【剑指Offer】搜索与回溯算法(1)
【剑指Offer】动态规划
【剑指Offer】动态规划(1)
【剑指Offer】动态规划(2)
【剑指Offer】双指针
【剑指Offer】双指针(1)
【剑指Offer】双指针(2)
【剑指Offer】搜索与回溯算法(2)
【剑指Offer】搜素与回溯算法(3)
【剑指Offer】排序
【剑指Offer】排序(1)