C++二叉树非递归统一形式

最新推荐文章于 2021-08-13 13:13:37 发布

JustSleep

最新推荐文章于 2021-08-13 13:13:37 发布

阅读量107

点赞数

分类专栏：树 C++ 文章标签：算法

本文链接：https://blog.csdn.net/qq_30122883/article/details/100171692

版权

树同时被 2 个专栏收录

17 篇文章 0 订阅

订阅专栏

C++

8 篇文章 0 订阅

订阅专栏

前序遍历：

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

中序遍历：

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

后序遍历：讲前序遍历的根左右变成根右左，然后反向插入，变成左右根。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s;
        TreeNode *p = root;
        while (!s.empty() || p) {
            if (p) {
                s.push(p);
                res.insert(res.begin(), p->val);
                p = p->right;
            } else {
                p = s.top(); s.pop();
                p = p->left;
            }
        }
        return res;
    }
};