二叉树的循环遍历和递归遍历(经典)

备注:以下代码均采用c++实现。

在这里插入图片描述

递归遍历

前序遍历:根、左、右

    // 前序遍历:根、左、右
    void beforeOrder(TreeNode* root, vector<int>& before){
        // 终止条件
        if(root == NULL){
            return;
        }
        // 放入当前节点
        before.push_back(root->val);
        // 放入左子节点
        beforeOrder(root->left, before);
        // 放入右子节点
        beforeOrder(root->right, before);
    }

中序遍历:左、跟、右

	// 中序遍历:左、根、右
    void middleOrder(TreeNode* root, vector<int>& middle){
        // 终止条件
        if(root == NULL){
            return;
        }
        // 放入左子节点
        middleOrder(root->left, middle);
        // 放入当前节点
        middle.push_back(root->val);
        // 放入右子节点
        middleOrder(root->right, middle);
    }

后序遍历:左、右、跟

 	// 后序遍历:左、右、根
    void afterOrder(TreeNode* root, vector<int>& after){
        // 终止条件
        if(root == NULL){
            return;
        }
        // 放入左子节点
        afterOrder(root->left, after);
        // 放入右子节点
        afterOrder(root->right, after);
        // 放入当前节点
        after.push_back(root->val);
    }

函数调用:

   /**
     * 
     * @param root TreeNode类 the root of binary tree
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > threeOrders(TreeNode* root) {
        // write code here
        vector<int> before;
        vector<int> middle;
        vector<int> after;
        vector<vector<int>> ret;
        if(root == NULL){
            ret.push_back(before);
            ret.push_back(middle);
            ret.push_back(after);
            return ret; 
        }
   
        beforeOrder(root, before);
        middleOrder(root, middle);
        afterOrder(root, after);
        
        ret.push_back(before);
        ret.push_back(middle);
        ret.push_back(after);
        return ret;
    }

循环遍历

前序遍历:跟、左、右

    // 前序遍历:根、左、右
    void beforeOrder(TreeNode* root, vector<int>& before){
        // 辅助栈
        stack<TreeNode*> s;
        
        // 前序遍历:根、左、右
        TreeNode* p = root; 
        s.push(p);
        while(!s.empty()){
            p = s.top();
            s.pop();
            before.push_back(p->val);
            if(p->right){
                s.push(p->right);
            }
            if(p->left){
                s.push(p->left);
            }
        }
    }

中序遍历:左、跟、右

    // 中序遍历:左、根、右
    void middleOrder(TreeNode* root, vector<int>& middle){
        // 辅助栈
        stack<TreeNode*> s;
        
        // 中序遍历:左、根、右
        TreeNode* p = root;
        p = root;
        while(!s.empty() || p){
            // 先找到最左边的节点
            while(p){
                s.push(p);
                p = p->left;
            }
            // 循环出栈
            if(!s.empty()){
                p = s.top();
                s.pop();
                // 左子节点为空,放入当前节点
                middle.push_back(p->val);
                // 遍历右节点
                p = p->right;
            }
        }
    }

后序遍历:左、右、跟

    // 后序遍历:左、右、根
    void afterOrder(TreeNode* root, vector<int>& after){
        
        // 辅助栈
        stack<TreeNode*> s;
        
        // 中序遍历:左、右、跟
        TreeNode* p = root;
        s.push(p);
        while(!s.empty()){
            p = s.top();
            s.pop();
            after.push_back(p->val);
            if(p->left){
                s.push(p->left);
            }
            if(p->right){
                s.push(p->right);
            }
        }
        // 翻转列表
        reverse(after.begin(), after.end());
    }

主函数调用

    /**
     * 
     * @param root TreeNode类 the root of binary tree
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > threeOrders(TreeNode* root) {
        // write code here
        vector<int> before;
        vector<int> middle;
        vector<int> after;
        vector<vector<int>> ret;
        if(root == NULL){
            ret.push_back(before);
            ret.push_back(middle);
            ret.push_back(after);
            return ret; 
        }
        
        // 递归方式调用
        before.clear();
        beforeOrder(root, before);
        middle.clear();
        middleOrder(root, middle);
        after.clear();
        afterOrder(root, after);
        
        ret.push_back(before);
        ret.push_back(middle);
        ret.push_back(after);
        return ret;
    }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值