将前中后序迭代法遍历整合成一种类似的代码

解决三个题目:前中后序的遍历
二叉树的前序遍历
题目:
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

二叉树的后序遍历
题目:
给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。

二叉树的中序遍历
题目:
给定一个二叉树的根节点 root ,返回它的 中序 遍历。
代码随想录
难以整合的原因
中序无法同时解决访问节点(遍历节点)和处理节点(将元素放进结果集)不一致的情况。
思路分为两部分
存储元素:右中左,出栈然后右中左然后一直这样,此时就是把左边的全部处理完(即先处理左树)。
回收元素:左边到顶了下一个为null了就回收元素,然后节点到中节点,此时也不进行右中左,节点到右节点时,此时继续进行右中左,右中左。(即将右节点当做根节点处理右树)。

解决访问节点(遍历节点)和处理节点(将元素放进结果集)不一致的情况:
从栈中取元素作为根节点,然后根据不同类型的跟节点进行不同的操作。
对左节点的操作以及对右节点的操作都是继续搜索(即继续右中左)对根节点的操作当然不需要右中左了(那么怎么判断到达根节点了呢,通过栈中根节点下一个元素为null来判断)

中序遍历代码如下:

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != NULL) {
                st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
                if (node->right) st.push(node->right);  // 添加右节点(空节点不入栈)

                st.push(node);                          // 添加中节点
                st.push(NULL); // 中节点访问过,但是还没有处理,加入空节点做为标记。

                if (node->left) st.push(node->left);    // 添加左节点(空节点不入栈)
            } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
                st.pop();           // 将空节点弹出
                node = st.top();    // 重新取出栈中元素
                st.pop();
                result.push_back(node->val); // 加入到结果集
            }
        }
        return result;
    }
};

前序遍历代码如下:
这就是按照右左中的顺序,第一个出栈即出中节点不进行右左中,当为左节点、右节点时继续进行右左中。

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != NULL) {
                st.pop();
                if (node->right) st.push(node->right);  // 右
                if (node->left) st.push(node->left);    // 左
                st.push(node);                          // 中
                st.push(NULL);
            } else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};

后续遍历代码如下:
这就是按照中右左的顺序,第一个第二个出栈分别为出左节点右节点都进行中右左的操作,当为中节点时不进行中右左。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != NULL) {
                st.pop();
                st.push(node);                          // 中
                st.push(NULL);

                if (node->right) st.push(node->right);  // 右
                if (node->left) st.push(node->left);    // 左

            } else {
                st.pop();
                node = st.top();
                st.pop();
                result.push_back(node->val);
            }
        }
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值