题解 | #二叉树的后序遍历#

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector
     */
    vector<int> postorderTraversal(TreeNode* root) {
        // write code here
        vector<int> res;
        stack<TreeNode*> s;
        //flag可以使用结构体实现，也可以用映射法表示
        stack<bool> flag;
        TreeNode* bt=root;
        while(s.empty()!=1||bt!=nullptr)
        {
            //遍历左子树
            while(bt!=nullptr)
            {
                s.push(bt);
                flag.push(false);//增加标志
                bt=bt->left;
            }
            
            //左子树遍历完成后，开始出栈
            while(s.empty()!=1&&flag.top()==true)
            {
                bt=s.top();
                s.pop();
                flag.pop();
                res.push_back(bt->val);
            }


            //注意这里，如果出栈发现bt==root,表明已经遍历结束。bt表示当前正在遍历的结点，如果不将其置空，将重新遍历树
            if(bt==root)
            {
                bt=nullptr;
            }

            //取元素遍历右子树，注意需要将此节点的标志改为true
            if(s.empty()!=1)
            {
                flag.top()=true;
                bt=s.top();//出栈后又入栈了
                bt=bt->right;
            }
        }

        return res;

    }
};