难度:困难
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3] 1 \ 2 / 3 输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
题目分析:
后序遍历非递归版,必须掌握基础知识。后序遍历是左右根,所以根会遍历两次,我们不能直接输出,这样的话根会重复输出两次。我们申明一个节点作为标志,相当于一个last_pMove,记录移动节点上一个节点,每次输出的时候需要判断右边节点是否为空,或者走过,如果走过那么输出,如果没有走过或者不为空,那么我们先记录该节点,然后往右走,接着走到最左边。以此类推。
参考代码:
/**
* Definition for a binary tree node.
* 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) {
vector<int> res;
if(root == NULL)
return res;
TreeNode* pMove = root;
stack<TreeNode*> node_stack;
TreeNode* flag_node = NULL;
//先走到最左边
while(pMove)
{
node_stack.push(pMove);
pMove = pMove->left;
}
while(!node_stack.empty())
{
pMove = node_stack.top();
node_stack.pop();
//判断右节点为空,或者走过右边了,那么根节点输出
if(pMove->right == NULL || pMove->right == flag_node)
{
res.push_back(pMove->val);
flag_node = pMove;
}else{
//如果右节点还没走过,那么我们往右走并记录
node_stack.push(pMove);
pMove = pMove->right;
//在走到最左边,边走边记录
while(pMove)
{
node_stack.push(pMove);
pMove = pMove->left;
}
}
}
return res;
}
};