二叉树|513.找树左下角的值、112. 路径总和、106.构造二叉树
513.找树左下角的值
自己审题思路
层序遍历找最后一层第一个元素
看完代码随想录题解后的收获
递归法的学习
代码:
class Solution {
public:
int findBottomLeftValue(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
int result = 0;
while (!que.empty()) {
int size = que.size();
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
if (i == 0) result = node->val; // 记录每一行第一个元素(循环覆盖,最终记录最后一行第一个元素)
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
}
return result;
}
};
代码(递归)
class Solution {
public:
int maxDepth = INT_MIN;
int result;
void traversal(TreeNode* cur, int depth) {
if(!cur->left && !cur->right) {
if(depth > maxDepth) {
maxDepth = depth; // 更新最大深度
result = cur->val; // 最大深度最左面的数值
}
return;
}
if(cur->left) traversal(cur->left, depth + 1);
if(cur->right) traversal(cur->right, depth + 1);
return;
}
int findBottomLeftValue(TreeNode* root) {
if(root == nullptr) return 0;
traversal(root, 0);
return result;
}
};
参考详解
112. 路径总和
自己审题思路
看完代码随想录题解后的收获
递归函数什么时候要有返回值,什么时候没有返回值(三点)
- 如果需要搜索整棵二叉树且不用处理递归返回值,递归函数就不要返回值。(113.路径总和ii)
- 如果需要搜索整棵二叉树且需要处理递归返回值,递归函数就需要返回值。 (236. 二叉树的最近公共祖先 )
- 如果要搜索其中一条符合条件的路径,那么递归一定需要返回值,因为遇到符合条件的路径了就要及时返回。(本题的情况)
代码:
class Solution {
public:
bool traversal(TreeNode* cur, int count) {
if(!cur->left && !cur->right && count == 0) return true;
if(!cur->left && !cur->right) return false;
if (cur->left) {
if(traversal(cur->left,count - cur->left->val)) return true;
}
if (cur->right) {
if(traversal(cur->right,count - cur->right->val)) return true;
}
return false;
}
bool hasPathSum(TreeNode* root, int targetSum) {
if(root == nullptr) return false;
return traversal(root, targetSum - root->val);
}
};
/**
* 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:
bool traversal(TreeNode* cur, int count) {
if(!cur->left && !cur->right && count == cur->val) return true;
if(!cur->left && !cur->right) return false;
if (cur->left) {
if(traversal(cur->left,count - cur->val)) return true;
}
if (cur->right) {
if(traversal(cur->right,count - cur->val)) return true;
}
return false;
}
bool hasPathSum(TreeNode* root, int targetSum) {
if(root == nullptr) return false;
return traversal(root, targetSum);
}
};
参考详解
构造二叉树
自己审题思路
这道题有思路:根据中序找到根节点,然后循环找,但是写代码的时候遇到不少问题。
看完代码随想录题解后的收获
理解思路和debug。
代码:
class Solution {
private:
TreeNode* traversal (vector<int>& inorder, vector<int>& postorder) {
if (postorder.size() == 0) return NULL;
// 后序遍历数组最后一个元素,就是当前的中间节点
int rootValue = postorder[postorder.size() - 1];
TreeNode* root = new TreeNode(rootValue);
// 叶子节点
if (postorder.size() == 1) return root;
// 找到中序遍历的切割点
int delimiterIndex;
for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
if (inorder[delimiterIndex] == rootValue) break;
}
// 切割中序数组
// 左闭右开区间:[0, delimiterIndex)
vector<int> leftInorder(inorder.begin(), inorder.begin() + delimiterIndex);
// [delimiterIndex + 1, end)
vector<int> rightInorder(inorder.begin() + delimiterIndex + 1, inorder.end() );
// postorder 舍弃末尾元素
postorder.resize(postorder.size() - 1);
// 切割后序数组
// 依然左闭右开,注意这里使用了左中序数组大小作为切割点
// [0, leftInorder.size)
vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
// [leftInorder.size(), end)
vector<int> rightPostorder(postorder.begin() + leftInorder.size(), postorder.end());
root->left = traversal(leftInorder, leftPostorder);
root->right = traversal(rightInorder, rightPostorder);
return root;
}
public:
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
if (inorder.size() == 0 || postorder.size() == 0) return NULL;
return traversal(inorder, postorder);
}
};
class Solution {
public:
TreeNode* dfs(vector<int>& inorder, int inStart, int inEnd, vector<int>& postorder, int postStart, int postEnd) {
if(inStart == inEnd ) return nullptr;
int rootValue = postorder[postEnd - 1];
int split;
for (int i = 0; i < inorder.size(); i++) {
if (rootValue == inorder[i]) {
split = i;
break;
}
}
TreeNode* root = new TreeNode(rootValue);
int inLeftStart = inStart;
int inLeftEnd = split;
int inRightStart = split + 1;
int inRightEnd = inEnd;
int postLeftStart = postStart;
int postLeftEnd = postLeftStart + (inLeftEnd - inLeftStart);
int postRightStart = postLeftEnd;
int postRightEnd = postEnd - 1;
root->left = dfs(inorder, inLeftStart, inLeftEnd, postorder, postLeftStart, postLeftEnd);
root->right = dfs(inorder, inRightStart, inRightEnd, postorder, postRightStart, postRightEnd);
return root;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
if(inorder.size() == 0 || postorder.size() == 0) return nullptr;
return dfs(inorder, 0, inorder.size(), postorder, 0, postorder.size());
}
};