C++版 - LeetCode 145: 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?

分析：

AC程式碼：

#include<iostream>

#include<vector>

#include<stack>

#include<algorithm>  // 引入 reverse函式，對vector進行反轉

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) {

        vector<int> res(0);

        if(root==NULL) return res;

        stack<TreeNode *> st;

        TreeNode *p=root;

        while(!st.empty()|| p!=NULL)

        {

            if(p!=NULL)

            {

            st.push(p);

            res.push_back(p->val);

            p=p->right;

            }

            if(p==NULL)

            {

                p=st.top();

                p=p->left;

                st.pop();

            }

        }

        reverse(res.begin(),res.end());    // 對向量及其內部結構進行反轉

     return res;

    }

};

// 以下為測試

int main()

{

	Solution sol;

	vector<int> res;

	TreeNode *root = new TreeNode(1);

    root->right = new TreeNode(2);

    root->right->left = new TreeNode(3); 

	res=sol.postorderTraversal(root);

	for(int i:res)

		cout<<i<<" ";  // 此處為vector遍歷的方法，C++11標準支援

	return 0;

}

另一解法(非遞迴)：

後序遍歷的麻煩在於訪問過右子樹之後，第二次訪問的時候就必須訪問根節點，而不是繼續訪問右子樹，所以使用pre來記錄上次訪問的節點...

class Solution {

public:

    vector<int> postorderTraversal(TreeNode *root) {

        vector<int> ans;

        if(root == NULL) return ans;

        stack<TreeNode* > s;

        TreeNode * p = root;

        TreeNode * pre = NULL;                 //  記錄上次訪問的節點

        while(p != NULL || !s.empty()) {

            while(p != NULL) {

                s.push(p);

                p = p->left;

            }

            if(!s.empty()) {

                p = s.top();

                s.pop();

                if(p->right == NULL || p->right == pre) {        // 右子樹為空，或者是訪問過

                    ans.push_back(p->val);

                    pre = p;                         // 記錄剛剛訪問的右節點

                    p = NULL;

                }

                else {

                    s.push(p);

                    p = p->right;

                }

            }

        }

        return ans;

    }

};