@代码随想录算法训练营第14天 | LeetCode144.二叉树的前序遍历,145.二叉树的后序遍历,94.二叉树的中序遍历 (递归加迭代法,以及统一迭代法)
144.二叉树的前序遍历
第一遍读题思考
递归想起来比较复杂,看一下视频理解一下。
代码随想录解法思路
非常的清楚。写递归代码的时候确定三件事1.确定传入参数和返回值2.确定终止条件3.确定单层递归的逻辑。
比如在遍历二叉树的时候,1.传入的参数就是当前节点和要返回的数组2.终止条件就是当前节点是null了,那么因为null我就不需要在进行左中右/中左右/左右中的赋值操作了所以就要终止。3.单层递归其实就是按顺序赋值,traversal函数本质就是在赋值,因为current永远是指向node的那node有left和right两个子节点,那么左右两个子节点会作为新的中间节点进行遍历,所以把当前node可以直接赋值到数组中,而左右子节点则需要调用自身函数进行赋值。
c++代码具体实现注意事项
需要知道二叉树自己应该怎么定义的.
(递归版本)
/**
* 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 traversal(TreeNode* cur, vector<int> &result){
if(cur==NULL){
return;
}
result.push_back(cur->val);
traversal(cur->left,result);
traversal(cur->right,result);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root, result);
return result;
}
};
(迭代版本)
/**
* 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:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if(root==NULL) return result;
st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
st.pop();
result.push_back(node->val);
if(node->right){
st.push(node->right);}
if(node->left){
st.push(node->left);}
}
return result;
}
};