LeetCode 144.二叉树的前序遍历
题目链接:144.二叉树的前序遍历
踩坑:先看了视频,所以没踩坑,但是需要注意的是不要陷入回溯法的陷阱。
思路:
- 递归:(1)确定递归函数的参数与返回值(2)确定返回停止条件(3)确定单层递归逻辑
- 迭代:用栈来模拟迭代过程。因为遍历的就是要处理的元素,处理完就要弹出,所以在弹出之前就需要把左右孩子都要入栈,同时考虑栈先入后处理的原则,应先压入右孩子,再压入左孩子。
代码:
/**
* 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 digui(TreeNode* root, vector<int>& v)
{
if(root == nullptr) return;
v.push_back(root->val);
digui(root->left, v);
digui(root->right, v);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
digui(root, result);
return result;
}
};
LeetCode 94.二叉树的中序遍历
题目链接:94.二叉树的中序遍历
踩坑:先看了视频,就没踩什么坑。
思路:
- 递归:调换单层递归逻辑的顺序即可。
- 迭代:这里与前序遍历不同,因为每次遍历到的根节点都是“中”,而需要处理的节点是“左”。所以逻辑成了如果当前节点非空,则将左节点压入栈。如果为空,则取栈顶元素处理并将右节点压入栈。这里不需要一次性将左右节点都压入栈,因为遍历的元素不会立马被弹出。
代码:
/**
* 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 digui(TreeNode* root, vector<int>& v)
{
if(root == nullptr) return;
digui(root->left, v);
v.push_back(root->val);
digui(root->right, v);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
digui(root, result);
return result;
}
};
LeetCode 145.二叉树的后序遍历
题目链接:145. 二叉树的后序遍历
踩坑:无
思路:
- 递归:调换单层递归逻辑的顺序即可
- 迭代:通过对前序遍历“中左右”的改动,即先压入左孩子,再压入右孩子,变成“中右左”,再对结果数组反转得到“左右中”即可。
代码:
/**
* 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 digui(TreeNode* root, vector<int>& v)
{
if(root == nullptr) return;
digui(root->left, v);
digui(root->right, v);
v.push_back(root->val);
}
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
digui(root, result);
return result;
}
};