1 二叉树的最大深度
题目
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
代码
//递归
class Solution {
public int maxDepth(TreeNode root) {
if(root==null) return 0;
int left=maxDepth(root.left);
int right=maxDepth(root.right);
return Math.max(left,right)+1;
}
}
//迭代
class Solution {
public int maxDepth(TreeNode root) {
if(root==null) return 0;
Queue<TreeNode> queue =new LinkedList<>();
int depth=0;
queue.offer(root);
while(!queue.isEmpty()){
int size=queue.size();
depth++;
while(size>0){
TreeNode temp=queue.poll();
if(temp.left!=null) queue.offer(temp.left);
if(temp.right!=null) queue.offer(temp.right);
size--;
}
}
return depth;
}
}
总结
递归是后序遍历
迭代是层序遍历
2 二叉树的最小深度
题目
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
代码
//递归
class Solution {
public int minDepth(TreeNode root) {
if(root==null) return 0;
int left=minDepth(root.left);
int right=minDepth(root.right);
if(root.left==null && root.right!=null)
return 1+right;
if(root.left!=null && root.right==null)
return 1+left;
return 1+Math.min(left,right);
}
}
//迭代
class Solution {
public int minDepth(TreeNode root) {
if(root==null) return 0;
Queue<TreeNode> queue =new LinkedList<>();
int depth=0;
queue.offer(root);
while(!queue.isEmpty()){
int size=queue.size();
depth++;
while(size>0){
TreeNode temp=queue.poll();
if(temp.left!=null) queue.offer(temp.left);
if(temp.right!=null) queue.offer(temp.right);
size--;
if(temp.left==null &&temp.right==null) return depth;
}
}
return depth;
}
}
总结
如果左子树为空,右子树不为空,说明最小深度是 1 + 右子树的深度。
反之,右子树为空,左子树不为空,最小深度是 1 + 左子树的深度。
最后如果左右子树都不为空,返回左右子树深度最小值 + 1 。
3 完全二叉树的结点个数
题目
给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
代码
//时间复杂度 O(n) 空间复杂度 O(n)
//递归
class Solution {
public int countNodes(TreeNode root) {
if(root==null) return 0;
int left=countNodes(root.left);
int right=countNodes(root.right);
return 1+left+right;
}
}
//迭代
class Solution {
public int countNodes(TreeNode root) {
if(root==null) return 0;
Queue<TreeNode> queue =new LinkedList<>();
int count=0;
queue.offer(root);
while(!queue.isEmpty()){
int size=queue.size();
while(size>0){
TreeNode temp=queue.poll();
count++;
if(temp.left!=null) queue.offer(temp.left);
if(temp.right!=null) queue.offer(temp.right);
size--;
}
}
return count;
}
}
//时间复杂度O(logn*logn) 空间复杂度O(logn)
class Solution {
public int countNodes(TreeNode root) {
if(root==null) return 0;
TreeNode leftNode=root.left;
TreeNode rightNode=root.right;
int leftdepth=0,rightdepth=0;
while(leftNode!=null){
leftdepth++;
leftNode=leftNode.left;
}
while(rightNode!=null){
rightdepth++;
rightNode=rightNode.right;
}
if(leftdepth==rightdepth)
return (2 << leftdepth) - 1;
return countNodes(root.left)+countNodes(root.right)+1;
}
}
总结
在完全二叉树中,如果递归向左遍历的深度等于递归向右遍历的深度,那说明就是满二叉树。满二叉树结点个数等于2^depth -1