二叉树非递归遍历

Copy from here

//
// Created by daery on 2021/8/1.
//

#include "search.h"
#include <stack>
#include <iostream>
#include <vector>
using namespace std;

//Binary Tree Node
typedef struct node
{
    int data;
    struct node* lchild;  //左孩子
    struct node* rchild;  //右孩子
}BTNode;

// 先序遍历
void PreOrderWithoutRecursion1(BTNode* root) {
    if (root == NULL) return;
    BTNode* p = root;
    stack<BTNode*> s;
    while (!s.empty() || p) { //边遍历边打印,并存入栈中,以后需要借助这些根节点(不要怀疑这种说法哦)进入右子树
        while (p) {
            cout << p->data;
            s.push(p);
            p = p->lchild;
        }
        //当p为空时,说明根和左子树都遍历完了,该进入右子树了
        if (!s.empty()) {
            p = s.top();
            s.pop();
            p = p->rchild;
        }
    }
    cout << endl;
}

//中序遍历
void InOrderWithoutRecursion1(BTNode* root){
    //空树
    if (root == NULL) return;
    //树非空
    BTNode* p = root;
    stack<BTNode*> s;
    while (!s.empty() || p) { //一直遍历到左子树最下边,边遍历边保存根节点到栈中
        while (p) {
            s.push(p);
            p = p->lchild;
        }
        //当p为空时,说明已经到达左子树最下边,这时需要出栈了
        if (!s.empty()) {
            p = s.top();
            s.pop();
            cout << p->data;
            //进入右子树,开始新的一轮左子树遍历(这是递归的自我实现)
            p = p->rchild;
        }
    }
}

//后序遍历
void PostOrderWithoutRecursion(BTNode* root) {
    if (root == NULL)
        return;
    stack<BTNode*> s;
    //pCur:当前访问节点,pLastVisit:上次访问节点
    BTNode* pCur, *pLastVisit;
    //pCur = root;
    pCur = root;
    pLastVisit = NULL;
    //先把pCur移动到左子树最下边
    while (pCur) {
        s.push(pCur);
        pCur = pCur->lchild;
    }
    while (!s.empty()) {
        //走到这里,pCur都是空,并已经遍历到左子树底端(看成扩充二叉树,则空,亦是某棵树的左孩子)
        pCur = s.top();
        s.pop();
        //一个根节点被访问的前提是:无右子树或右子树已被访问过
        if (pCur->rchild == NULL || pCur->rchild == pLastVisit) {
            cout << pCur->data;
            //修改最近被访问的节点
            pLastVisit = pCur;
        }
        /*这里的else语句可换成带条件的else if:
		else if (pCur->lchild == pLastVisit)//若左子树刚被访问过,则需先进入右子树(根节点需再次入栈)
		因为:上面的条件没通过就一定是下面的条件满足。仔细想想!
		*/
        else {
            //根节点再次入栈
            s.push(pCur);
            //进入右子树,且可肯定右子树一定不为空
            pCur = pCur->rchild;
            while (pCur) {
                s.push(pCur);
                pCur = pCur->lchild;
            }
        }
    }
    cout << endl;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值