试解leetcode算法题--二叉树的后序遍历

<题目表述>

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]

1
 \
  2
 /
3

输出: [3,2,1]
进阶: 递归算法很简单，你可以通过迭代算法完成吗？

<原题链接>

https://leetcode-cn.com/problems/binary-tree-postorder-traversal

<思路>

递归算法仅需要几句即可写出，在这里我们之间按照进阶要求来写非递归算法。

对于了解数据结构的人来说，本题目只需要写出 后序非递归 算法即可，并没有很大的难度。

<样例代码>

#include<stack>
#include<vector>
#include<iostream>
using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> sta;
        vector<int> ans;
        TreeNode * p = root;
        TreeNode * q = NULL;
        while(p != NULL || !sta.empty())
        {
            if(p != NULL)
            {
                sta.push(p);
                p = p->left;
            }
            else
            {
                p = sta.top();

                if(p->right != NULL && p->right != q)
                    p = p->right;
                else
                {
                    sta.pop();
                    ans.push_back(p->val);
                    q = p;
                    p = NULL;
                }
            }
        }
        return ans;
    }
};