【LeetCode】110.平衡二叉树
题意:给定一个二叉树,判断它是否是平衡二叉树.
思路:递归后序,返回当前传入节点为根节点的树的高度。左右中,中的高度就是根的高度。
代码:
/**
* 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 getHeight(TreeNode* root)
{
if (!root)
{
return 0;
}
int leftHeight = getHeight(root->left);
if (-1 == leftHeight)
{
return -1;
}
int rightHeight = getHeight(root->right);
if (-1 == rightHeight)
{
return -1;
}
if (abs(rightHeight - leftHeight) > 1)
{
return -1;
}
else
{
return max(rightHeight, leftHeight) + 1;
}
}
bool isBalanced(TreeNode* root) {
return getHeight(root) == -1 ? false : true;
}
};
【LeetCode】257. 二叉树的所有路径
题意:给你一个二叉树的根节点 root ,按 任意顺序 ,返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。
思路:前序遍历,求路径需要回溯,也就是递归到叶子节点时把路径记录一下。
代码:
/**
* 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:
void getPath(TreeNode* root, vector<int> &path, vector<string> &result)
{
path.push_back(root->val);
if (!root->left && !root->right)
{
string res;
for (int i = 0; i < path.size(); ++i)
{
if (0 != i)
{
res += "->";
}
res += to_string(path[i]);
}
result.push_back(res);
return ;
}
if (root->left)
{
getPath(root->left, path, result);
path.pop_back();
}
if (root->right)
{
getPath(root->right, path, result);
path.pop_back();
}
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<int> path;
vector<string> result;
getPath(root, path, result);
return result;
}
};
【LeetCode】404.左叶子之和
题意:给定二叉树的根节点 root ,返回所有左叶子之和。
思路:后序遍历,左叶子结点的计算是在父节点上,递归时要注意根的左叶子之和就是左子树的左叶子和右子树的左叶子之和。
代码:
/**
* 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)
{
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);
return leftSum + rightSum;
}
};
心态:“第六章 二叉树 part04” 拿下!
参考资料