递归
每次写递归,都按照这三要素来写,可以保证大家写出正确的递归算法!
确定递归函数的参数和返回值: 确定哪些参数是递归的过程中需要处理的,那么就在递归函数里加上这个参数, 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。
确定终止条件: 写完了递归算法, 运行的时候,经常会遇到栈溢出的错误,就是没写终止条件或者终止条件写的不对,操作系统也是用一个栈的结构来保存每一层递归的信息,如果递归没有终止,操作系统的内存栈必然就会溢出。
确定单层递归的逻辑: 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。
迭代
后序遍历的迭代写法
后序非递归遍历二叉树,先访问左子树,再访问右子树,最后访问根节点。具体步骤为:
- 沿根的左孩子依次入栈,直到左孩子为空
- 读栈顶元素
2.1 若右孩子不为空或右孩子未被访问过,转为右孩子结点继续执行1
2.2 否则,栈顶元素出栈并访问
为记录元素是否被访问,设立pre指针,指向上一个出栈元素。若当前执行节点p的右孩子已被访问,说明上一个出栈元素正是p的右孩子。若未被访问则pre != p->right
代码
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> ans;
if(root == nullptr)return {};
TreeNode* p = root;
TreeNode* pre = nullptr; //pre指向最近出栈的结点
while(p || !st.empty()){
if(p != nullptr){//不能写成p->left != nullptr然后入栈左子树,应该先入栈p本身
st.push(p);
p = p->left;
// pre = root;
}else{
//不能直接跟if-else,此时遍历到最左子树时p是nullptr
p = st.top();
if(p->right && pre != p->right){
p = p->right;
// pre = root; 这里还没有出栈所以不能加
}else{//出栈
TreeNode* tmp = st.top();
ans.emplace_back(tmp->val);
st.pop();
pre = tmp;
p = nullptr;
}
}
}
return ans;
}
};
每次出栈访问完一个结点就相当于遍历完以该结点为根的树,所以需要将p置为null(重新取栈顶)
标签:pre,遍历,出栈,迭代,递归,nullptr,st,二叉树 From: https://www.cnblogs.com/neromegumi/p/18534641