二叉树遍历代码

最新推荐文章于 2024-01-25 21:49:27 发布

3ilence

最新推荐文章于 2024-01-25 21:49:27 发布

阅读量278

点赞数

分类专栏：笔记

本文链接：https://blog.csdn.net/qq_43556545/article/details/109095347

版权

笔记专栏收录该内容

127 篇文章 0 订阅

订阅专栏

本文回顾了二叉树的遍历，特别是迭代方式的前序遍历代码，适合数据结构复习。

摘要由CSDN通过智能技术生成

今天复习了一下二叉树的遍历，当初学数据结构的时候没有好好理解，迭代的代码如下：
//前序

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> s;
        s.push(root);
        while(!s.empty()){
            TreeNode* r=s.top();
            s.pop();
            if(!r) continue;
            ans.push_back(r->val);
            s.push(r->right);
            s.push(r->left);     
        }
        return ans;
    }
};

//中序

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> stk;
        while (root != nullptr || !stk.empty()) {
            while (root != nullptr) {
                stk.push(root);
                root = root->left;
            }
            root = stk.top();
            stk.pop();
            res.push_back(root->val);
            root = root->right;
        }
        return res;
    }
};

//后序

 class Solution { 
 public:
       vector<int> postorderTraversal(TreeNode *root) {
           vector<int> res;
           if (root == nullptr) {
               return res;
           }
   
       stack<TreeNode *> stk;
       TreeNode *prev = nullptr;
       while (root != nullptr || !stk.empty()) {
           while (root != nullptr) {
               stk.emplace(root);
               root = root->left;
           }
           root = stk.top();
           stk.pop();
           if (root->right == nullptr || root->right == prev) {
               res.emplace_back(root->val);
               prev = root;
               root = nullptr;
           } else {
               stk.emplace(root);
               root = root->right;
           }
       }
       return res; } };*