一、平衡二叉树 (优先掌握递归)
题目一:110.平衡二叉树 (优先掌握递归)
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
isBalanced
函数检查树是否平衡。它调用checkBalance
函数,并检查返回值是否为-1(表示不平衡)。
checkBalance
函数递归地检查每个节点。如果一个节点的左右子树高度差超过1,或者其任一子树不平衡,则该树不平衡。如果一个节点的左右子树都是平衡的,并且左右子树的高度差不超过1,那么这个节点是平衡的,函数返回这个节点的高度(即左右子树的最大高度加1)。
public:
bool isBalanced(TreeNode* root) {
return checkBalance(root) != -1;
}
private:
int checkBalance(TreeNode* node) {
if (node == nullptr) {
return 0;
}
int leftHeight = checkBalance(node->left);
if (leftHeight == -1) return -1; // 左子树不平衡
int rightHeight = checkBalance(node->right);
if (rightHeight == -1) return -1; // 右子树不平衡
if (abs(leftHeight - rightHeight) > 1) return -1; // 当前节点不平衡
return max(leftHeight, rightHeight) + 1; // 返回节点的高度
}
二、二叉树的所有路径
题目一:257. 二叉树的所有路径
binaryTreePaths
函数初始化一个空的字符串向量,然后调用findPaths
函数。
findPaths
函数递归地遍历树。它将当前节点的值追加到路径字符串中。如果当前节点是叶子节点(即没有子节点),则将当前路径添加到结果向量中。对于非叶子节点,函数递归地调用自身来处理左子树和右子树,每次调用时将当前节点的值和一个箭头符号 "->" 追加到路径字符串中。
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> paths;
if (root != nullptr) {
findPaths(root, "", paths);
}
return paths;
}
private:
void findPaths(TreeNode* node, string path, vector<string>& paths) {
if (node->left == nullptr && node->right == nullptr) {
paths.push_back(path + to_string(node->val));
return;
}
if (node->left != nullptr) {
findPaths(node->left, path + to_string(node->val) + "->", paths);
}
if (node->right != nullptr) {
findPaths(node->right, path + to_string(node->val) + "->", paths);
}
}
三、左叶子之和
题目一:404. 左叶子之和
给定二叉树的根节点 root
,返回所有左叶子之和。
如果当前节点是
nullptr
,直接返回0。检查当前节点的左子节点是否存在且是叶子节点(即没有左右子节点)。如果是,将其值加到总和中。
递归地对左右子树调用
sumOfLeftLeaves
函数,并将返回的值加到总和中。返回计算的总和。
/*
* @lc app=leetcode.cn id=404 lang=cpp
*
* [404] 左叶子之和
*/
// @lc code=start
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if (root == nullptr) {
return 0;
}
int sum = 0;
if (root->left != nullptr && root->left->left == nullptr && root->left->right == nullptr) {
sum += root->left->val;
}
sum += sumOfLeftLeaves(root->left);
sum += sumOfLeftLeaves(root->right);
return sum;
}
};
// @lc code=end