平衡二叉树
- 题目:Leetcode110
- 思路:
- 递归:根据平衡二叉树定义,一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1,明确了要比较两棵子树的高度,使用后序遍历。标记左右子树是否差值大于1,如果已经不是二叉平衡树了,可以返回-1 来标记已经不符合平衡树的规则了。判断当前节点的二叉树是否为平衡二叉树,分别求出左右子树的高度,若高度差小于等于1,则返回当前二叉树的高度,否则返回-1,表示已经不是二叉平衡树。
class Solution {
public:
//返回以该节点为根节点的二叉树的高度,如果不是平衡二叉树了则返回-1
int getHeight(TreeNode* node) {
if(node == nullptr) return 0;
int leftHeight = getHeight(node->left);
if(leftHeight == -1) return -1;
int rightHeight = getHeight(node->right);
if(rightHeight == -1) return -1;
return abs(rightHeight - leftHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
}
bool isBalanced(TreeNode* root) {
return getHeight(root) == -1 ? false : true;
}
};
二叉树的所有路径
- 题目:Leetcode257
- 思路:题目要求从根节点到叶子的路径,所以需要前序遍历(递归)。在遍历过程中,涉及到回溯,要把已经走过的路径记录下来,通过回溯回退到前面的节点,再进入另一条路径。因为是前序遍历,需要先处理中间节点,中间节点就是我们要记录路径上的节点;然后是递归和回溯的过程,递归前要加上判断语句,下面要递归的节点是否为空;递归完要进行回溯,回溯和递归是一一对应的,有一个递归,就要有一个回溯。
class Solution {
public:
void traversal(TreeNode* cur, vector<int>& path, vector<string>& res) {
//前序遍历
path.push_back(cur->val);//中,中为什么写在这里,因为最后一个节点也要加入到path中
if(cur->left == nullptr && cur->right == nullptr) {//终止条件:如果cur为叶子结点
string sPath;
for(int i = 0; i < path.size() - 1; i++) {
sPath += to_string(path[i]);
sPath += "->";
}
sPath += to_string(path[path.size() - 1]);//单独处理最后一个结点
res.push_back(sPath);
return;
}
if(cur->left) {//左
//回溯和递归是一一对应的,有一个递归,就要有一个回溯
traversal(cur->left, path, res);
path.pop_back();//回溯
}
if(cur->right) {//右
traversal(cur->right, path, res);
path.pop_back();//回溯
}
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<int> path;
vector<string> res;
if(root == nullptr) return res;
traversal(root, path, res);
return res;
}
};
左叶子之和
- 题目:Leetcode404
- 思路:
- 注意辨析左叶子和左侧节点的区别。节点A的左孩子不为空,且左孩子的左右孩子都为空(说明是叶子节点),那么A节点的左孩子为左叶子节点。因此,判断是否为左叶子节点,必须通过该节点的父节点来判断:判断父节点的左孩子是不是左叶子。
- 递归的遍历顺序为后序遍历(左右中),是因为要通过递归函数的返回值来累加求取左叶子数值之和。当遇到左叶子节点的时候,记录数值,然后通过递归求取左子树左叶子之和,和 右子树左叶子之和,相加便是整个树的左叶子之和。
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
//只有当前遍历的节点是父节点,才能判断其子节点是不是左叶子
//如果当前遍历的节点是叶子节点,那其左叶子也必定是0
if(root == nullptr) return 0;
if(!root->left & !root->right) return 0;//如果遍历到叶子节点直接返回,可以少一层迭代
int leftSum = sumOfLeftLeaves(root->left);//左
//判断当前节点的左孩子就是一个左叶子的情况
if(root->left && !root->left->left && !root->left->right) {
leftSum = root->left->val;//记录数值
}
int rightSum = sumOfLeftLeaves(root->right);//右,求右子树左叶子之和
int sum = leftSum + rightSum;//中,将左子树左叶子之和与右子树左叶子之和相加
return sum;
}
};
总结
注意高度与深度、左叶子与左侧节点的概念辨析以及相应的处理逻辑。
回溯和递归是一一对应的,有一个递归,就要有一个回溯。