图解二叉树非递归版的前序遍历算法

“ 图解用栈数据结构对树的前序遍历，中序遍历，后续遍历。”

树的遍历

所谓遍历 (Traversal) 是指沿着某条搜索路线，依次对树中每个结点均做一次且仅做一次访问。访问结点所做的操作依赖于具体的应用问题。

二叉树遍历

本文以二叉树的遍历为例总结。

二叉树由根结点及左、右子树这三个基本部分组成。因此，在任一给定结点上，可以按某种次序执行三个操作：

访问结点本身（Node）
遍历该结点的左子树（Left subtree）
遍历该结点的右子树（Right subtree）

根据访问结点操作发生位置不同，命名：

前序遍历 Preorder Traversal ：访问根结点的操作发生在遍历其左右子树之前。
中序遍历 Inorder Traversal：访问根结点的操作发生在遍历其左右子树之中间。
后序遍历 Postorder Traversal ： 访问根结点的操作发生在遍历其左右子树之后。

二叉树的数据结构定义如下所示：

    public class TreeNode<T>
    {
        public T val { get; set; }
        public TreeNode<T> left { get; set; }
        public TreeNode<T> right { get; set; }
        public TreeNode(T data)
        {
            val = data;
        }
    }

思考算法

我们借助栈，先将根节点推入栈内，因为首先访问根节点，这意味着它必须先出栈，但是我们还没有机会让左右子树入栈呢？

自然地，我们想到定义一个临时变量引用根节点，这样在循环时，不断修改这个临时变量，我们在此处将它成为上下文 context，它始终指向当前遍历子树的根节点，如下图所示，假设为某个遍历时的子树，此时context指向的便是子树的根节点，节点1出栈后，通过context引用它，然后找到它的左右子树，即 context.left, context.right。

想明白这个context的作用，栈s的出栈时机，进栈时机在前序遍历不难理解，所以循环体内完成一个完整的一步做的事情也就容易理解了，一个遍历步骤做的事情如下：

1. context指向的当前子树的根节点为null吗？ 如果是，则将context指向栈 s的栈顶，即下一个子树根节点，实际上为右子树。
2. 访问栈顶，并出栈栈顶元素，出栈了不用担心，我们通过context引用了这个出栈元素。
3. context.right 进栈
4. context.left 进栈，至于为什么是先进栈right，这里不解释了。
5. 既然访问完了子树的根节点，那么前序遍历下一步是访问左子树了，自然地，context被迭代为context.left 。

前序遍历（栈结构实现）

前序遍历 Preorder Traversal ：先访问根结点，然后访问根节点的左子树，最后访问根节点的右子树。

给定如下图所示的二叉树，

那么用栈这种数据结构，如何前序遍历这颗二叉树呢？

首先，我们把代码放在这里，然后分析这个代码是怎么想出来的。

        public static IList<T> PreorderTraversal<T>(
                                   TreeNode<T> root)
        {
            IList<T> rtn = new List<T>();
            if (root == null) return rtn;
            var s = new Stack<TreeNode<T>>();
            s.Push(root);
            TreeNode<T> context = root;
            while (s.Count > 0) {           
                if (context == null)
                    context = s.Peek();
                rtn.Add(s.Peek().val); //访问栈顶元素
                s.Pop();
                if (context.right != null) //左子树入栈
                    s.Push(context.right);
                if (context.left != null)  //右子树入栈
                    s.Push(context.left);
                context = context.left;
            }
            return rtn;
        }

while遍历前，各个变量的状态
s 栈有一个节点 1
context 上下文变量引用节点 1

第一次遍历，此时栈中含有元素，进入循环体内，上下文为节点 1，不满足 if 条件，访问 节点 1，节点 1 出栈， 节点 1 的右子树不为 null 先入栈，左子树也为 null后入栈， 上下文变量赋值为 左子树节点 2，如下图所示：

context = node2

第二次遍历，访问节点 2，然后节点 2 出栈，节点 2 的右子树不为 null，入栈，此时的 context 被赋值为节点 2的左子树，即为 null

context = null

第三次遍历，会特殊一些了，因为此时的context为 null，注意源码实现中的 if 条件，
if (context == null)
context = s.Peek();
满足了条件，context 被赋值为栈顶节点 4，然后执行逻辑与以上遍历相同，访问栈顶节点 4，显然这个节点 4的左右子树皆为 null，context再次被赋值，且等于 null。

context = null

第四次遍历，依然满足 if 条件，context 被赋值为栈顶节点 3，执行逻辑与上相似，访问栈顶节点 3，节点 3 出栈，节点 3 的右子树为null，左子树不为null，入栈，context再次被赋值为节点 3 的左子树 5

context = node5

第五次遍历，不满足 if 条件了，直接访问节点 5， 然后节点 5 出栈，显然节点 5 左右子树为null，context赋值为node5.left，即为null，并且栈的元素已经空了。

context = null

此时，while (s.Count > 0) 不满足了，退出遍历。

前序遍历总结

前序遍历在遍历前先入栈根节点，context引用根节点，进入遍历后，有一个 if 条件判断context为null吗，如果是说明上文的左子树为null，context需要引用此时栈顶元素，访问栈顶元素，栈顶元素出栈，接下来，首先入栈右子树，然后入栈左子树，context为 context.left 进行迭代。因此，以上二叉树访问节点的次序为

1->2->4->3->5

中序遍历

中序遍历，遍历根节点位于遍历左子树和右子树之间，即左子树，根节点，右子树。

中序遍历的源码如下：

  public IList<T> InorderTraversal<T>(TreeNode<T> root)
        {
            IList<T> rtn = new List<T>();
            var s = new Stack<TreeNode<T>>();
            if (root == null) return rtn;
            s.Push(root);
            while (s.Count > 0)
            {
                var context = s.Peek();
                while (context != null) 
                {
                    s.Push(context.left);
                    context = context.left;
                }
                s.Pop();
                if (s.Count == 0)
                    return rtn;
                rtn.Add(s.Peek().val); 
                TreeNode<T> curNode = s.Pop();
                s.Push(curNode.right);
            }
            return rtn;
        }

不妨您先思考一下，这段代码是怎么构思出来的，分析中序遍历，后续遍历图解，敬请关注明天的推送，谢谢。

本文由算法实验室倾情奉献，除合作社区及合作媒体外，禁止转载。
算法实验室分享最新的算法和应用实现相关技术内容。