LeetCode 94 二叉树中序遍历 递归非递归 通俗易懂的万能模板方法

LeetCode 94 二叉树中序遍历 递归非递归 通俗易懂的万能模板方法


二叉树的遍历方式作为面试热门考点,相信递归的方式大家都很容易能够写出来,但只掌握递归的方法是不够的,还需要的掌握非递归的方式,介绍一种能够解决前、中、后三种遍历的非递归形式的模板。一次性解决非递归问题!!

前序遍历的万能模板法
后序遍历的万能模板法

1.递归 二叉树中序遍历

中序遍历的顺利为 左子树,根节点,右子树。递归主要就是递归结构和递归结束条件。

递归结构:那么递归的顺序就是 先遍历左子树,保存根节点,遍历右子树。
递归结束条件:因为没有明确的结束条件,因此递归条件就是遇到NULL,返回即可。

class Solution {
    
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        func(root,res);
        return res;        
    }
    
    void func(TreeNode* root, vector<int>& res)
    {
        if(root==NULL)	 递归结束条件
            return ;
        func(root->left,res);		 遍历左子树
        res.push_back(root->val);    保存根节点
        func(root->right,res);		 遍历右子树
    }
};

2.非递归 二叉树中序遍历 万能模板法

对于非递归法,难度会高一些,我们将递归的方式通过使用的数据结构来模拟递归。

对于万能模板法,同样也是借用栈的数据结构,同时我们引入了一个类,来提供额外的信息。

我们引入了Guide类。

class Guide{ 
public:
    int ope;
    TreeNode* node;
    Guide(int ope,TreeNode* node):ope(ope),node(node){}  
};

我们在使用栈的过程,不存储树节点,而是存储Guide类的对象,Guide类对象包含了树节点以及额外的信息(int ope),实际上ope只有两种取值,0和1。
ope = 0代表了我们首次遍历到了某个树节点(我们首次遍历的某个节点会有三种类型的操作)

  1. 将当前节点的值存入最后结果中 (也就是将ope赋值为1)。
  2. 遍历左子树。
  3. 遍历右子树。

对于中序遍历是左,根,右。那也存入栈中顺序为右,根,左。对应的核心代码

st.push(Guide(0,s.node->right));
st.push(Guide(1,s.node));
st.push(Guide(0,s.node->left));

ope = 1代表了我们将此对象包含的树节点的值存入到最终的vector中。

那么最终的完整版代码如下。对于前序,后序非递归遍历的只需要改动上述三条语句的顺序。

class Guide{
public:
    int ope;
    TreeNode* node;
    Guide(int ope,TreeNode* node):ope(ope),node(node){}    
};

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        if(root == NULL)
            return res;
        stack<Guide> st;
        st.push(Guide(0,root));
        while(!st.empty())
        {
            Guide s = st.top();
            st.pop();
            if(s.node == NULL)
                continue;
            if(s.ope == 1)
            {
                res.push_back(s.node->val);
            }else{
                st.push(Guide(0,s.node->right));
                st.push(Guide(1,s.node));
                st.push(Guide(0,s.node->left));
            }
        }  
        return res;
    }    
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值