二叉树的前序中序后序遍历-非递归-使用同一段代码实现

树的遍历通常使用递归，因为它的实现更简单，代码也更容易理解。

但在面试，或者特殊的情境中会使用到迭代算法（非递归）。

此时需要使用栈去模拟函数栈调用过程。

 

本文将给出一段代码去实现这三种遍历

相比于传统的方式：前序遍历，中序遍历，后序遍历，使用不同的方式代码去实现，并且后续遍历更为难理解一些

可拓展性更好（比如N叉树的遍历），也更容易理解

 

考虑，对于一个函数栈，它除了存储了一些变量和指令，同时还存储了当前执行位置。

对于树的遍历，无非为：t->val，t->left ，t->right 三个代码的排列

因此，我们只需定义一个int类型的变量，用于记录当前执行到哪一个代码。

    class TreeNodeWithIndex{
        public TreeNode node;
        public int index=0;
        public TreeNodeWithIndex(TreeNode n) { node = n; }
    }
　　　　//0 pre ; 1 inorder ; 2 post 
    public IList<int> TreeTraversal(TreeNode root,int type)
    {
        var stack = new Stack<TreeNodeWithIndex>();
        var ans = new List<int>();
        stack.Push(new TreeNodeWithIndex(root));
        while (stack.Count!=0)
        {
            var i = stack.Peek();
            if (i.index > 2 || i.node == null) { stack.Pop(); continue; }
            if (type == i.index) ans.Add(i.node.val);
            else
            {
                int tmp = type - i.index * 2;
                if (tmp > 0 || tmp == -2)
                    stack.Push(new TreeNodeWithIndex(i.node.left));
                else
                    stack.Push(new TreeNodeWithIndex(i.node.right));
            }
            i.index++;
        }
        return ans;
    }