栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

递归与栈(二叉树前中后遍历)

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

递归与栈(二叉树前中后遍历)

 写法一:递归

class Solution {
public:
    void preorder(TreeNode* root, vector &res) {
        if (root == NULL) {
            return;
        }
        res.push_back(root->val);
        preorder(root->left, res);
        preorder(root->right, res);
    }
    vector preorderTraversal(TreeNode* root) {
        vector res;
        preorder(root, res);
        return res;
    }
};

写法二:栈

class Solution {
public:
    vector preorderTraversal(TreeNode* root) {
        stack st;
        vector res;
        if (root == NULL) return res;
        st.push(root);
        while (!st.empty()) {
            TreeNode* temp = st.top();
            st.pop();
            res.push_back(temp->val);
            if (temp->right) st.push(temp->right);
            if (temp->left) st.push(temp->left);
        }
        return res;
    }
};

通过这两种方法可以发现,递归的底层其实是栈,递归的过程就是每次递归会把函数的变量,参数值,返回地址压到栈中,等递归返回的时候,再把栈顶弹出去。感觉就很奇妙!

class Solution {
public:
    vector inorderTraversal(TreeNode* root) {
        stack st;
        vector res;
        TreeNode* cur = root;
        while (cur != nullptr || !st.empty()) {
            if (cur != nullptr) {
                st.push(cur);
                cur = cur->left;    //左
            } else {
                cur = st.top();      //中
                res.push_back(cur->val);
                st.pop();
                cur = cur->right;           //右
            }
        }
        return res;
    }
};

 中序遍历有点特殊,需要一个指针,先指到最左端,当cur为null时,让cur为栈顶元素然后输出其值,相当于已经到了“中”这个地步,然后向右走一格。

 

class Solution {
public:
    vector postorderTraversal(TreeNode* root) {
        stack st;
        vector result;
        if (root == NULL) return result;
        st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            st.pop();
            result.push_back(node->val);
            if (node->left) st.push(node->left); // 相对于前序遍历,这更改一下入栈顺序 (空节点不入栈)
            if (node->right) st.push(node->right); // 空节点不入栈
        }
        reverse(result.begin(), result.end()); // 将结果反转之后就是左右中的顺序了
        return result;
    }
};

其实后序遍历就是改了一下先序遍历,因为先序遍历是中左右,我先把他变为中右左,然后再把他反转,变为左右中,则是后序遍历。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/872996.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号