二叉树的递归遍历和迭代遍历

递归

每次写递归,都按照这三要素来写,可以保证大家写出正确的递归算法!

确定递归函数的参数和返回值: 确定哪些参数是递归的过程中需要处理的,那么就在递归函数里加上这个参数, 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。

确定终止条件: 写完了递归算法, 运行的时候,经常会遇到栈溢出的错误,就是没写终止条件或者终止条件写的不对,操作系统也是用一个栈的结构来保存每一层递归的信息,如果递归没有终止,操作系统的内存栈必然就会溢出。

确定单层递归的逻辑: 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。

迭代

后序遍历的迭代写法

后序非递归遍历二叉树,先访问左子树,再访问右子树,最后访问根节点。具体步骤为:

  1. 沿根的左孩子依次入栈,直到左孩子为空
  2. 读栈顶元素
    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(重新取栈顶)

posted @   NeroMegumi  阅读(26)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示