首页 > 代码库 > leetcode 刷题之路 78 Binary Tree Postorder Traversal

leetcode 刷题之路 78 Binary Tree Postorder Traversal

Given a binary tree, return the postorder traversal of its nodes‘ values.

For example:
Given binary tree {1,#,2,3},

   1
         2
    /
   3

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?

非递归方式实现二叉树的后序遍历。

思路:使用栈辅助实现,首先将根节点入栈,然后以栈是否为空为条件进行循环操作,取栈顶元素:

当栈顶元素为叶子节点(栈顶元素的左右孩子节点都是NULL)或者其左右孩子已经遍历过时(previous等于左孩子节点或者右孩子节点),则出栈并记录到vi数组中,同时使将该节点赋给previous指针,本次遍历的节点作为下一次判断的上一个节点。

如果以上条件都不满足,说明栈顶节点的孩子节点还没有遍历到,那么将左右孩子节点入栈,为保证出栈时左孩子节点在右孩子节点前面,应该先将右孩子入栈。

AC code:

/**
 * Definition for binary tree
 * 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> vi;
        if(root==NULL)
            return vi;
        TreeNode *previous=NULL;
        stack<TreeNode*> st;
        st.push(root);
        while(!st.empty())
        {
            root=st.top();
        if((root->left==NULL&&root->right==NULL)||((previous!=NULL&&previous==root->left)||(previous!=NULL&&previous==root->right)))
                {
                    vi.push_back(root->val);
                    st.pop();
                    previous=root;
                }
                else
                {
                    if(root->right!=NULL)
                        st.push(root->right);
                    if(root->left!=NULL)
                        st.push(root->left);
                }
            
        }
        return vi;
    }
};