一、平衡二叉树-LeetCode 110
Leecode链接: leetcode 977
文章链接: 代码随想录
视频链接: B站
给定一个二叉树,判断它是否是
平衡二叉树
示例:
输入:root = [3,9,20,null,null,15,7]
输出:true
思路
判断是否是平衡的,需要根据左右子树的高度差来判断,因为涉及到高度,所以函数返回值需要int型,在遇到左右子树高度差大于1时该返回-1,表示不平衡。
实现代码
//cpp
class Solution {
public:
// 返回以该节点为根节点的二叉树的高度,如果不是平衡二叉树了则返回-1
int getHeight(TreeNode* node) {
if (node == NULL) {
return 0;
}
int leftHeight = getHeight(node->left);
if (leftHeight == -1) return -1;
int rightHeight = getHeight(node->right);
if (rightHeight == -1) return -1;
return abs(leftHeight - rightHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
}
bool isBalanced(TreeNode* root) {
return getHeight(root) == -1 ? false : true;
}
};
个人问题
有思路没实现代码,个人觉得涉及高度问题,左右子树必须返回高度值,这样才能判断是否平衡,但不平衡时该返回什么,当时没有想清楚,只觉得应该返回bool值,但返回bool值怎么判断高度差,最终导致代码没有写出来。
二、二叉树的路径-LeetCode 257
Leecode链接: LeetCode 257
文章链接: 代码随想录
视频链接: B站
给你一个二叉树的根节点 root ,按 任意顺序 ,返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。
示例:
输入:root = [1,2,3,null,5]
输出:["1->2->5","1->3"]
思路
使用回溯加递归,遍历时使用一个path数组记录遍历到的数字,遍历到叶子节点就插入到res数组中记录下来,然后回退path。
实现代码
//cpp
class Solution {
private:
void traversal(TreeNode* cur, string path, vector<string>& result) {
path += to_string(cur->val); // 中,中为什么写在这里,因为最后一个节点也要加入到path中
if (cur->left == NULL && cur->right == NULL) {
result.push_back(path);
return;
}
if (cur->left) {
path += "->";
traversal(cur->left, path, result); // 左
path.pop_back(); // 回溯 '>'
path.pop_back(); // 回溯 '-'
}
if (cur->right) {
path += "->";
traversal(cur->right, path, result); // 右
path.pop_back(); // 回溯'>'
path.pop_back(); // 回溯 '-'
}
}
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
string path;
if (root == NULL) return result;
traversal(root, path, result);
return result;
}
};
个人问题
有思路但没写出代码。
总结
题目需要注意插入节点值这个操作需要放最上面,因为退出条件是遍历到叶子节点就插入res数组,这样可以避免没有插入path数组就直接将结果放到了res数组中;值拷贝的原因在于回溯思想,这样可以避免来回对path操作。
三.左叶子之和-LeeCode 404
Leecode链接: LeetCode 404
文章链接: 代码随想录
视频链接: B站
给定二叉树的根节点 root ,返回所有左叶子之和。
示例:
输入: root = [3,9,20,null,null,15,7]
输出: 24
解释: 在这个二叉树中,有两个左叶子,分别是 9 和 15,所以返回 24
思路
使用后序遍历,计算各个子树中的左叶子之和,然后相加。
实现代码
//cpp
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if(root == nullptr){
return 0;
}
int leftsum = sumOfLeftLeaves(root->left);
if(root->left && root->left->left == nullptr && root->left->right == nullptr){
leftsum = root->left->val;
}
int rightsum = sumOfLeftLeaves(root->right);
int sum = leftsum + rightsum;
return sum;
}
};
个人问题
代码没有自己写出来。
总结
需要注意的是,本体不仅需要判断是不是叶子节点还要判断是不是左叶子节点,所以不能用当前节点来判断,而需要借助父节点来判断。时间复杂度O(n^2),空间复杂度O(1)。