非递归遍历二叉树

后续遍历暂时没写,留作后期补充

#include <iostream>
#include <stack>

struct Node {
    int val;
    struct Node *left{nullptr}, *right{nullptr};
    Node (int n = 0): val(n), left(nullptr), right(nullptr) {}
};

inline void visit(Node* &node) {
    std::cout << node->val << " ";
}

// 前序递归创建二叉树
Node* PreOrderCreate() {
    int n;
    std::cin >> n;
    if (n < 0) {
        return nullptr;
    }
    auto node = new Node(n);
    node->left = PreOrderCreate();
    node->right = PreOrderCreate();
    return node;
}

// 前序遍历
void StackPreOrder(Node* root) {
    if (root == nullptr) {
        return;
    }
    std::stack<Node*> st;
    st.push(root);
    auto p = root;
    while (!st.empty()) {  // 外层函数递归栈
        p = st.top();
        st.push(p->right); // 放在栈中,模拟函数压栈
        st.push(p->left);
    }
}

// 中序遍历
void StackInOrder(Node* root) {
    if (root == nullptr) {
        return;
    }
    std::stack<Node*> st;
    auto p = root;
    st.push(p);
    do {
        while (p->left != nullptr) {
            p = p->left;
            st.push(p);
        }
        p = st.top();
        st.pop();
        visit(p);
        if (p->right != nullptr) {
            st.push(p->right);
        } 
    } while (!st.empty());
}

// 后序遍历
void StackPostOrder(Node* root) {
    if (root == nullptr) {
        return;
    }
    std::stack<Node*> st;
    auto p = root;
    st.push(p);
    do {
        while (p->left != nullptr) {
            p = p->left;
            st.push(p);
            if (p->right != nullptr) {
                st.push(p->right);
            }
        }
        p = st.top();
        visit(p);
        st.pop();
    } while (!st.empty());
}

int main() {
    auto root = PreOrderCreate();
    std::cout << "PreOrder:" << std::endl;
    StackPreOrder(root);
    std::cout << "InOrder:" << std::endl;
    StackInOrder(root);
    std::cout << "PostOrder:" << std::endl;
    StackPostOrder(root);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值