110 平衡二叉树
题目链接:
110. 平衡二叉树 - 力扣(LeetCode)
给定一个二叉树,判断它是否是平衡二叉树。
方法1 自顶向下
先序遍历
class Solution {
public:
int depth(TreeNode* root)
{
if(root == nullptr) return 0;
return max(depth(root->left), depth(root->right)) + 1;
}
bool isBalanced(TreeNode* root) {
if(root == nullptr) return true;
return abs(depth(root->left) - depth(root->right)) <=1 && isBalanced(root->left) && isBalanced(root->right);
}
};
方法2 自底向上
参考题解:
代码随想录 (programmercarl.com)
后序遍历
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;
}
};
257 二叉树的所有路径
题目链接:257. 二叉树的所有路径 - 力扣(LeetCode)
给你一个二叉树的根节点 root
,按 任意顺序 ,返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。
思路:
先序遍历,碰到叶子节点,放入到ans中。
class Solution {
public:
void dfs(TreeNode* cur, string path, vector<string>& ans)
{
if(cur == nullptr)
{
return;
}
if(cur->left == nullptr && cur->right == nullptr)
{
if(path.empty()) path += to_string(cur->val);
else
{
path += "->";
path += to_string(cur->val);
}
ans.push_back(path);
}
if(path.empty()) path += to_string(cur->val);
else
{
path += "->";
path += to_string(cur->val);
}
dfs(cur->left, path, ans);
dfs(cur->right, path, ans);
return;
}
vector<string> binaryTreePaths(TreeNode* root) {
string path;
vector<string> ans;
dfs(root, path, ans);
return ans;
}
};
代码可精简:
参考:代码随想录 (programmercarl.com)
class Solution {
public:
void traversal(TreeNode* cur, string path, vector<string>& result) {
path += to_string(cur->val); // 中
if (cur->left == NULL && cur->right == NULL) {
result.push_back(path);
return;
}
if (cur->left) traversal(cur->left, path + "->", result); // 左
if (cur->right) traversal(cur->right, path + "->", result); // 右
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
string path;
if (root == NULL) return result;
traversal(root, path, result);
return result;
}
};
404 左叶子之和
题目链接:
404. 左叶子之和 - 力扣(LeetCode)
给定二叉树的根节点 root
,返回所有左叶子之和。
思路:
对每个节点添加一个状态state,设定根节点state=-1,左孩子state=1,右孩子state=0,当为叶子节点,且state为1时,加入到ans中。
class Solution {
public:
void dfs(TreeNode* cur, int state, int& ans)
{
if(cur == nullptr) return;
if(cur->left == nullptr && cur->right == nullptr)
{
if(state == 1) ans += cur->val;
return;
}
dfs(cur->left, 1, ans);
dfs(cur->right, 0, ans);
return;
}
int sumOfLeftLeaves(TreeNode* root) {
int ans = 0;
int state = -1;
dfs(root, state, ans);
return ans;
}
};