104. 二叉树的最大深度
题目
给定一个二叉树 root
,返回其最大深度。
二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。
示例:
输入:root = [3,9,20,null,null,15,7] 输出:3
思路
用递归来做,其中心思路是,一个结点的最大深度相当于其左右子树的最大深度加一,这就可以利用递归求得子树深度了。
接下来是递归三部曲:
1. 首先确定返回值和参数,返回值肯定是深度,参数则是二叉树结点;
2. 其次确定递归终止条件,也就是当结点为空时返回0;
3. 最后明确每一次递归要做的事,其实就是按找中心思路返回最大深度。
代码
class Solution {
public:
int getdepth(TreeNode* node) {
if (node == NULL) return 0;
int leftdepth = getdepth(node->left);
int rightdepth = getdepth(node->right);
int depth = 1 + max(leftdepth, rightdepth);
return depth;
}
int maxDepth(TreeNode* root) {
return getdepth(root);
}
};
111. 二叉树的最小深度
题目
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
思路
这道题的中心思路跟上面的一样,都是用递归法,每次取左右子树最小深度加一,不过这里有一个易错点,那就是深度要从叶子结点开始算,所以当遇到一个只有一个子树的结点时,不能记录空的一边,而是递归返回有子树的那边的深度。
代码
class Solution {
public:
int getdepth(TreeNode* node) {
if (node == NULL) return 0;
int leftdepth = getdepth(node->left);
int rightdepth = getdepth(node->right);
if (node->left == NULL && node->right != NULL)
return 1 + rightdepth;
if (node->left != NULL && node->right == NULL)
return 1 + leftdepth;
int depth = 1 + min(leftdepth, rightdepth);
return depth;
}
int minDepth(TreeNode* root) {
return getdepth(root);
}
};
222. 完全二叉树的结点个数
题目
给你一棵 完全二叉树 的根节点 root
,求出该树的节点个数。
完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h
层,则该层包含 1~ 2h
个节点。
普通解法
管他完不完全,反正也是二叉树,既然是二叉树,之前学过的递归和非递归遍历都可以用来求结点数,只需要把原来的结点值入vector变成计数器加一,下面是用非递归前序遍历来做的:
class Solution {
public:
int countNodes(TreeNode* root) {
stack<TreeNode*> st;
int num = 0;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* cur = st.top();
st.pop();
num++;
if (cur->right) st.push(cur->right);
if (cur->left) st.push(cur->left);
}
return num;
}
};
完全二叉树解法
完全二叉树可以不断拆分为一个满二叉树和一个完全二叉树,每一个满二叉树可以一边往下走一边求深度,如果左右子树的深度相同,就可以直接计算出这个二叉树的结点个数,因为满二叉树的结点数等于2的深度次方减一,如果不是满二叉树,就递归求解这个完全二叉树的根节点的左右子树的结点数再加一。
class Solution {
public:
int countNodes(TreeNode* root) {
if (root == NULL) return 0;
TreeNode* left = root->left;
TreeNode* right = root->right;
int leftDepth = 0, rightDepth = 0;
while (left) {
left = left->left;
leftDepth++;
}
while (right) {
right = right->right;
rightDepth++;
}
if (leftDepth == rightDepth) {
return (2 << leftDepth) - 1;
}
return countNodes(root->left) + countNodes(root->right) + 1;
}
};