【数据结构】二叉树四种遍历的非递归算法

递归算法虽然简洁,但一般而言,其执行效率不高。因此,有时候要把递归算法转化为非递归算法。

先序遍历非递归算法

由先序遍历过程可知,先访问根节点,再访问左子树,最后访问右子树。因此,先将根节点进栈,在栈不空时循环如下:p出栈,访问*p结点,若右孩子不空将该右孩子结点进栈,若左孩子不空再将左孩子结点进栈。对应算法如下:

template <typename T>
void PreOrder(BTNode<T> * root) {
    BTNode<T> * St[MAX], *p;
    int top = -1;
    if (root != NULL) {
        top++;
        St[top] = root;
        while (top > -1) {
            p = St[top];
            top--;
            cout << p->data << " ";
            if (p->rchild != NULL) {
                top++;
                St[top] = p->rchild;
            }
            if (p->lchild != NULL) {
                top++;
                St[top] = p->lchild;
            }
        }
        cout << endl;
    }
}

中序遍历非递归算法

由中序遍历的过程可知,中序序列的开始结点是一棵二叉树的最左下结点,其基本思路是先找到二叉树的开始结点,访问它,在处理其右子树。由于二叉树链中指针的链接是单向的,因此采用一个栈保存需要返回的结点指针。其算法过程是:用指针指向当前要处理的结点。先扫描(并非访问)根结点的所有左结点并将他们一一进栈,当无左结点时表示栈顶结点无左子树,然后出栈这个结点,并访问它,将p指向刚出栈结点的右孩子,对右孩子进行同样的处理。需要注意的是,当结点*p的所有左下结点进栈后,这时的栈顶结点要么没有左子树要么其左子树已经访问过,就可以访问这个栈顶结点。如此这样,直到栈空为止。对应的算法如下:

template<typename T>
void InOrder(BTNode<T> * root) {
    BTNode<T> *St[MAX], *p;
    int top = -1;
    if (root != NULL) {
        p = root;
        while (top > -1 || p != NULL) {     //处理*root结点的左子树
            while (p != NULL) {             //扫描*p的所有左结点并进栈
                top++;
                St[top] = p;
                p = p->lchild;
            }
            //执行到此处,栈顶元素没有左孩子或左孩子树均已访问过
            if (top > -1) {
                p = St[top];                //出栈*p结点
                top--;
                cout << p->data << " ";     //访问数据
                p = p->rchild;              //转向处理*p的右孩子结点
            }
        }
        cout << endl;
    }
}

后序遍历非递归算法

在后序遍历中,对于每个结点,先访问其左子树,然后访问右子树,最后才是该结点本身。与中序遍历情况类似,后序遍历中第一个访问的结点是二叉树的最左下结点。由于首先访问结点的左右子树,而后才访问结点本身,所以必须告知地点,其左右子树是否访问过。

采用一个栈保存需要返回的结点指针,先扫描根结点的所有左结点并一一进栈,出栈一个结点*root作为当前结点,然后扫描该结点的右子树。当一个结点的左右孩子结点均访问后再访问该结点,如此循环直到栈空为止。

其中的难点是如何判断一个结点*root的右子树已访问过(实际上当root结点的右孩子结点已访问过的话,则其右子树就已访问过),为此用p保存刚刚访问过的结点(初值为NULL),若root->rchild == p成立(在后序遍历中,root的右孩子结点一定刚好在root之前访问过),说明*root的左右子树均已访问,现在应访问*root。

从上述过程可知,栈中保存的是当前结点*root的所有祖先结点(均未被访问过)。对应的算法如下:

template<typename T>
void PostOrder(BTNode<T> * root) {
    BTNode<T> *St[MAX], *p;
    int top = -1, flag;
    if (root != NULL) {
        do {
            while (root != NULL) {      //将*root所有左结点入栈
                top++;
                St[top] = root;
                root = root->lchild;
            }
            p = NULL;                   //p指向当前结点的前一个已访问的结点
            flag = 1;
            while (top != -1 && flag) {
                root = St[top];             //取出当前的栈顶元素
                if (root->rchild == p) {
                    cout << root->data << " ";
                    top--;
                    p = root;           //指向被访问的结点
                } else {
                    root = root->rchild;
                    flag = 0;
                }
            }
        } while (top != -1);
        cout << endl;
    }
}

层序遍历算法

在进行层序遍历时,对某一层的结点访问完后,再按照他们的访问次序对各个结点的左孩子和右孩子顺序访问,这样一层一层进行,先访问的结点其左右孩子也要先访问,这与队列的操作原则比较吻合。因此,在进行层序遍历时可设置一个队列存放已访问的结点。遍历从二叉树的根结点开始,首先将根指针入队,然后从队头取出一个元素,每取一个元素,执行下面的操作:
访问该结点;
若该结点的左右孩子非空,则将其左孩子和右孩子指针入队;
此过程不断进行,当队列为空时,二叉树的层次遍历结束。

template<typename T>
void LeverOrder(BTNode<T> * root) {
    int front;
    int rear = 0;
    BTNode<T> * Q[MAX];
    BTNode<T> * q;
    if (root == NULL) {
        return;
    } else {
        Q[rear++] = root;
        while (front != rear) {
            q = Q[front++];
            cout << q->data << " ";
            if (q->lchild != NULL) Q[rear++] = q->lchild;
            if (q->rchild != NULL) Q[rear++] = q->rchild;
        }
    }
}
    原文作者:递归算法
    原文地址: https://blog.csdn.net/shujh_sysu/article/details/52156714
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞