二叉树的递归和非递归遍历

前段时间瞄了一眼《剑指offer》上面有说要熟悉二叉树的6种遍历方式，说来你或许不信，我自己实现非递归遍历的时候硬是没有实现出来，看来自己的基本功还是有待加强的。话不多说，上代码加分析。

二叉树的实现：

这个基本上是轻车熟路了，平常写书的算法的时候都要求实现一遍，当然这里不是这篇文章的重点。

树的节点的定义和树的定义。

struct TreeNode {    TreeNode *lChild;  //左孩子    TreeNode *rChild; //右孩子    EleType value;       //节点中的值};struct Tree {    TreeNode *root;   //树可以直接用一个节点的指针来表示};

树的实现。

TreeNode *CreateTreeNode (EleType k) {  //这个函数的意义是分配一个树的节点并且初始化    TreeNode *temp = (TreeNode *) malloc (sizeof (TreeNode));    temp->lChild = nullptr;    temp->rChild = nullptr;    temp->value = k;    return temp;}void CreateTree (TreeNode **root) {    int t;    cin >> t;    if (t != 0) {   //为零的话代表其是叶节点        *root = CreateTreeNode (t);        CreateTree (&((*root)->lChild));//利用递归实现树        CreateTree (&((*root)->rChild));    }}

二叉树的前序遍历：

二叉树的前序遍历的递归实现。

void PreTravelTree (TreeNode *node) {    if (node != nullptr) {        cout << node->value << " ";        PreTravelTree (node->lChild);        PreTravelTree (node->rChild);    }}

二叉树的前序遍历的非递归实现。

void PreTravelTreeNonRecur (TreeNode *node) {    stack<TreeNode *> nodeStack;    TreeNode *cur = node;    while (cur != nullptr || !nodeStack.empty ()) {        while (cur != nullptr) {            cout << cur->value << " ";            nodeStack.push (cur);            cur = cur->lChild;        }        if (!nodeStack.empty ()) {            cur = nodeStack.top ();            nodeStack.pop ();            cur = cur->rChild;        }    }

二叉树的中序遍历：

二叉树的中序遍历的递归实现。

void MidTravelTree (TreeNode *node) {    if (node != nullptr) {        MidTravelTree (node->lChild);        cout << node->value << " ";        MidTravelTree (node->rChild);    }}

二叉树的中序遍历的非递归实现。

void MidTravelTreeNonRecur (TreeNode *node) {    stack<TreeNode *> nodeStack;    TreeNode *cur = node;    while (cur != nullptr || !nodeStack.empty ()) {        while (cur != nullptr) {            nodeStack.push (cur);            cur = cur->lChild;        }        if (!nodeStack.empty ()) {            cur = nodeStack.top ();            cout << cur->value << " ";            nodeStack.pop ();            cur = cur->rChild;        }    }}

二叉树的后序遍历：

二叉树的后序遍历的递归实现。

void PosTravelTree (TreeNode *node) {    if (node != nullptr) {        PosTravelTree (node->lChild);        PosTravelTree (node->rChild);        cout << node->value << " ";    }}

二叉树的后序遍历的非递归实现。

这个是这篇文章的重点了，由于二叉树的后序遍历不同于前序遍历和中序遍历，必须要使用某种方式标记出左右字节点是否已经访问过了，那么在这里其实有两种方式，一种方式是改变底层的数的节点的数据结构，加上访问标记。第二种就是使用不同的进栈方式，在父节点进栈的时候同时将两个子节点进栈。下面先介绍第一种方式，改变底层的数据结构。

（底层数据结构只是加上了visited的字段，并没有其他的不同，不再赘述）

void PosTravelTreeNonRecurSec (TreeNodePlus *node) {    stack<TreeNodePlus *> nodeStack;    TreeNodePlus *cur = node;    while (cur != nullptr || !nodeStack.empty ()) {        while (cur != nullptr) {  //在这个while循环当中，只将不是nullptr同时没有访问过的节点加入nodeStack            cur->visited++;            nodeStack.push (cur);            if (cur->lChild == nullptr) {                break;            }else if (cur->lChild->visited > 0) {                break;            }else {                cur = cur->lChild;            }        }        if (!nodeStack.empty ()) { //进入这里就说明其子节点的一个孩子或者两个孩子已经访问过了（这里有一个特殊情况            cur = nodeStack.top ();//就是子节点为nullptr也可以进入这里，但是子节点为nullptr也代表访问过了）            cur->visited++;            if (cur->visited == 3) {  //说明这个节点的左右子节点已经访问过了，那么就可以将这个节点打印出来                cout << cur->value << " ";                nodeStack.pop ();                cur = nullptr;  //将其设置为nullptr那么下个循环就可以跳过上面那么while循环，处理这个节点的父节点。            }else {                cur = cur->rChild;//说明这个节点的右节点还没有访问，那么把当前节点的值设置为右节点。            }        }    }}

第二种方式就是使用不同的进栈方式。

void PosTravelTreeNonRecur (TreeNode *node) {    stack<pair<TreeNode * , bool>> nodeStack; //这个栈的节点是一个pair    TreeNode *cur = node;    nodeStack.push (make_pair (cur , false));    bool visited;    while (!nodeStack.empty ()) {        cur = nodeStack.top().first;        visited = nodeStack.top().second;        nodeStack.pop ();        if (cur == nullptr)            continue;        if (visited)            cout << cur->value << " ";        else {  //使用精心设计的进栈方式，根节点放在最下面，然后再是右节点，再是坐节点，这样就保证取出来的时候            nodeStack.push (make_pair(cur , true));  //左节点可以最先取出来。同时，其他两种的遍历方式的非递归实现 nodeStack.push (make_pair(cur->rChild , false)); //也可以采用这种进栈方式，但是缺点是效率稍微低了一些，进栈            nodeStack.push (make_pair(cur->lChild , false)); //次数多了。        }    }}