【数据结构】二叉树遍历

先序遍历和中序遍历非递归代码：

#include <iostream>
#include <vector>
using namespace std;

typedef struct BinaryTree 
{
    int data;
    struct BinaryTree *rchild, *lchild;
}BinaryTree;

int createBinaryTree( BinaryTree * &T)  //必须用引用 因为内存是在函数里面分配的
{
        int ch;
        scanf("%d", &ch);

        if(ch != 0)
        {
            T = (BinaryTree *)malloc(sizeof(BinaryTree));
            T->data = ch;
            createBinaryTree(T->lchild);
            createBinaryTree(T->rchild);
        }
        else
        {
            T = NULL;
        }
    return 0;
}

int visit(int data)
{
    printf("%d ", data);
    return 0;
}


int PreOrderTraverse(BinaryTree T) //先序遍历  
{
    BinaryTree* p;
    vector<BinaryTree *> Stack;
    Stack.push_back(&T);
    while(!Stack.empty())
    {
        while((p = Stack.back()) != NULL) 
        {
            visit(p->data);
            Stack.push_back(p->lchild);
        }
        Stack.pop_back();
        if(!Stack.empty())
        {
            p = Stack.back(); 
            Stack.pop_back();
            Stack.push_back(p->rchild);
        }
    }
    return 0;
}


int InOrderTraverse(BinaryTree T) //中序遍历
{
    BinaryTree* p;
    vector<BinaryTree *> Stack;
    Stack.push_back(&T);
    while(!Stack.empty())
    {
        while((p = Stack.back()) != NULL) 
            Stack.push_back(p->lchild);
        Stack.pop_back();
        if(!Stack.empty())
        {
            p = Stack.back(); 
            Stack.pop_back();
            visit(p->data);
            Stack.push_back(p->rchild);
        }

    }
    return 0;
}

int main()
{
    BinaryTree * T = NULL;
    createBinaryTree(T);
    PreOrderTraverse(*T);
    InOrderTraverse(*T);
    return 0;
}

注意理清楚弹栈的机制。

---------------------------------

今天参考了一下别人的思路 补上后序遍历非递归算法 注意如何设置在右子树弹出时双亲结点的访问和弹出

int AferOrderTraverse(BinaryTree T) //后序遍历
{
    BinaryTree * p;
    vector<BinaryTree *> Stack;
    int tag[30] = {0}; //用tag标签记录存入的是左子树0 还是右子树1
    int tagnum = 0; //记录存放的数量
    Stack.push_back(&T);
    tag[0] = 0;
    tagnum = 1;

    while(!Stack.empty())
    {
        while((p = Stack.back())!= NULL)
        {
            Stack.push_back(p->lchild);
            tag[tagnum++] = 0;
        }
        //只要右子树弹出，其双亲结点就被访问并弹出
        while(tag[tagnum - 1] == 1)  //若是右子树 先弹出开始的空集 访问其parent结点（一定是其上一个结点） 再循环弹出
        {
            Stack.pop_back();
            tagnum--;
            visit(Stack.back()->data);
        }
        Stack.pop_back();
        tagnum--;
    
        if(!Stack.empty())
        {
            p = Stack.back();
            Stack.push_back(p->rchild);
            tag[tagnum++] = 1;
        }
    }
    return 0;
}