用递归和非递归的方法实现二叉树遍历

用递归和非递归的方法实现二叉树遍历。

递归中序遍历：

struct TreeNode
{
    int val;
    TreeNode* left;
    TreeNode* right;
};

void InOrderRec(TreeNode* root)
{
    if (root == NULL)
        return;

    if (root->left)
        InOrderRec(root->left);

    cout << root->val;

    if (root->right)
        InOrderRec(root->right);
}

非递归中序遍历：
使用栈来保存中间节点。
1.先将根节点压栈。
2.将左子节点压栈，直至左子节点不存在，此时位于栈顶的应该是最左侧的根节点。
3.将栈顶的结点打印并弹出。
4.将当前节点设为右子节点，返回1循环。

void InOrderUnRec(TreeNode * root)
{
    if (root == NULL)
        return;

    stack<TreeNode*> s;
    TreeNode* node = root;

    while (node != NULL || !s.empty())
    {
        while (node != NULL)
        {
            s.push(node);
            node = node->left;
        }

        if (!s.empty())
        {
            node = s.top();
            s.pop();
            cout << node->val << endl;
            node = node->right;
        }
    }
}

递归方法实现另外两种遍历相对简单，这里就不赘述。着重介绍非递归方法。

非递归方法实现前序遍历：
使用栈保存中间节点。对于任一节点node
1.将node输出，然后压栈。
2.判断node的左子树存在，若不存在，则取出栈顶元素并出栈，同时将栈顶元素的右子节点设为node，压栈，执行1的操作。若不为空，则将左子树赋值为node。
3.直到node为空并且栈为空，遍历结束。

void InOrderUnRec(TreeNode * root)
{
    if (root == NULL)
        return;

    stack<TreeNode*> s;
    TreeNode* node = root;

    while (node != NULL || !s.empty())
    {
        while (node != NULL)
        {
            cout<<node->val<<endl;
            s.push(node);
            node = node->left;
        }

        if (!s.empty())
        {
            node = s.top();
            s.pop();
            node = node->right;
        }
    }
}

非递归方法实现后序遍历：
后序遍历相对于前面两个比较难。对任意节点，要先访问其左子节点再访问其右子节点。因此，要设置一个标志位，表示右子节点是否被访问过。
对任意一个节点node，初始化flag=false，表示第一次访问，即右子节点还没有访问过。
1.将node压入栈。
2.判断左子节点是否存在：如果存在，则将node赋值为左子节点，循环至1的操作；若不存在，则判断node的flag是否为false，是false，将node赋值为右子节点同时将flag赋值为true，循环至1的操作，若是true，表示右子节点已经遍历过，则输出node，将栈顶元素弹出，node赋值为null。
3.直至node为空并且栈为空，遍历结束。

struct TreeNodeWithFlag
{
    int val;
    TreeNodeWithFlag* left;
    TreeNodeWithFlag* right;
    bool visited;
};
void PostOrderUnRec(TreeNodeWithFlag* root)
{
    if (root == NULL)
        return;
    TreeNodeWithFlag* node = root;

    stack<TreeNodeWithFlag*> s;

    while (node != NULL || !s.empty())
    {
        while (node)
        {
            s.push(node);
            node = node->left;
        }

        if (!s.empty())
        {
            node = s.top();
            if (node->visited == false)
            {
                node->visited = true;
                node = node->right;
            }
            else
            {
                cout << node->val << " ";
                s.pop();
                node = NULL;
            }
        }
    }
}