代码随想录算法训练营第15天 | 110.平衡二叉树、257.二叉树的所有路径、404.左叶子之和、222.完全二叉树的节点个数
文章目录
110.平衡二叉树
解题思路
首先要了解平衡二叉树的定义:每个节点的左子树和右子树的高度差的绝对值不超过1。
本题使用递归法来解决,递归三部曲:
- 首先确定递归函数的参数和返回值:参数是传入的节点指针,返回值是以该节点为根节点的二叉树的高度
- 确定终止条件:遇到空节点终止
- 确定单层递归的逻辑:要看左子树高度和右子树高度的高度差,求出左右子树的高度,若差值小于等于1,则返回当前二叉树的高度,否则返回-1。
代码实现
class Solution {
public:
int getDepth(TreeNode* node) {
if (node == nullptr) {
return 0;
}
int leftDepth = getDepth(node->left);
if (leftDepth == -1)
return -1;
int rightDepth = getDepth(node->right);
if (rightDepth == -1)
return -1;
return abs(leftDepth - rightDepth) > 1 ? -1
: 1 + max(leftDepth, rightDepth);
}
bool isBalanced(TreeNode* root) {
return getDepth(root) == -1 ? false : true;
}
};
题目总结
当我们既需要求二叉树的高度,又需要返回一个结果(确定他处于某种状况),如是不是平衡二叉树,就可以多返回一个定值,如-1,通过这个值就可以做到两者同时判断。
257.二叉树的所有路径
解题思路
本题要求从根节点到叶子节点的所有路径,需要使用前序遍历,方便父节点指向子节点,找到对应的路径,本题中会涉及到回溯,因为需要记录路径,需要回溯操作回退一条路径从而进入另一个路径。
递归三部曲:
- 递归函数参数和返回值:传入二叉树根节点同时要用两个数组记录每条路径 path 和最终结果集 result
- 终止条件:当 cur 不为空且其左右孩子均为空时
- 单层递归逻辑:因为是前序遍历,所以需要处理中间节点,放进 path 中,接着便开始递归和回溯
代码实现
class Solution {
public:
void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
path.push_back(cur->val);
if (cur->left == nullptr && cur->right == nullptr) {
string sPath;
for (int i = 0; i < path.size() - 1; i++) {
sPath += to_string(path[i]);
sPath += "->";
}
sPath += to_string(path[path.size() - 1]);
result.push_back(sPath);
return;
}
if (cur->left) {
traversal(cur->left, path, result);
path.pop_back();
}
if (cur->right) {
traversal(cur->right, path, result);
path.pop_back();
}
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
vector<int> path;
if (root == nullptr)
return result;
traversal(root, path, result);
return result;
}
};
题目总结
注意使用两个数组,一个记录路径,另一个为最终结果;另外,还要注意将回溯写入递归里,能够将每次存到值减少,从而得到所有路径。
404.左叶子之和
解题思路
递归三部曲:
- 递归函数的参数和返回值:传入根节点,返回值是数值之和
- 终止条件:遍历到空节点
- 单层递归逻辑:遇到左边叶子节点,记录数值,递归求取左子树左边叶子节点和右子树左叶子节点相加就是左叶子之和
代码实现
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if (root == nullptr)
return 0;
if (root->left == nullptr && root->right == nullptr)
return 0;
int leftValue = sumOfLeftLeaves(root->left);
if (root->left && !root->left->left && !root->left->right) {
leftValue = root->left->val;
}
int rightValue = sumOfLeftLeaves(root->right);
int sum = leftValue + rightValue;
return sum;
}
};
题目总结
平时解二叉树的题目时,已经习惯了通过节点的左右孩子判断本节点的属性,而本题我们要通过节点的父节点判断本节点的属性,需要注意。
222.完全二叉树的节点个数
解题思路
本题我选择使用的是用普通二叉树的解法
递归三部曲:
- 递归函数的参数和返回值:参数是根节点,返回就返回以该节点为根节点二叉树的节点数量,所以返回值为int类型。
- 终止条件:空节点
- 单层递归逻辑:求左右子树节点数量,再加上根节点
代码实现
class Solution {
public:
int getNodesNum(TreeNode* cur) {
if (cur == NULL)
return 0;
int leftNum = getNodesNum(cur->left);
int rightNum = getNodesNum(cur->right);
int result = leftNum + rightNum + 1;
return result;
}
int countNodes(TreeNode* root) { return getNodesNum(root); }
};
题目总结
遍历顺序是左右中,后序遍历。