树的三种遍历方式

1.中序遍历思路：每到一个节点 A，因为根的访问在中间，将 A 入栈。然后遍历左子树，接着访问 A，最后遍历右子树。

#include <iostream>
#include <vector>
#include <stack>

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

class Solution
{
public:
	std::vector<int> inorderTraversal(TreeNode* root)
	{
		std::stack<TreeNode*> st;
		std::vector<int> vec;
		TreeNode* rt = root;

		while(rt || st.size())
		{
			while(rt)
			{
				st.push(rt);
				rt = rt->left;
			}

			rt = st.top();
			st.pop();
			vec.push_back(rt->val);
			rt = rt->right;
		}
	}
};

2.先序遍历思路：先树根，然后左子树，然后右子树。每棵子树递归。
在迭代算法中，思路演变成，每到一个节点 A，就应该立即访问它。
因为，每棵子树都先访问其根节点。对节点的左右子树来说，也一定是先访问根。
在 A 的两棵子树中，遍历完左子树后，再遍历右子树。因此，在访问完根节点后，遍历左子树前，要将右子树压入栈。

#include <iostream>
#include <vector>
#include <stack>

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
class Solution
{
public:
	std::vector<int> preorderTraversal(TreeNode* root)
	{
		std::stack<TreeNode*> st;
		std::vector<int> vec;
		TreeNode* rt = root;

		while(rt || st.size())
		{
			while(rt)
			{
				st.push(rt->right);
				vec.push_back(rt->val);
				rt = rt->left;
			}
			rt = st.top();
			st.pop();
		}
		return vec;
	}
};

3.后序遍历
（1）
我们可以用与前序遍历相似的方法完成后序遍历。后序遍历与前序遍历相对称。思路： 每到一个节点 A，就应该立即访问它。 然后将左子树压入栈，再次遍历右子树。遍历完整棵树后，结果序列逆序即可（使用反转迭代器）

#include <iostream>
#include <vector>
#include <stack>

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
class Solution
{
public:
	std::vector<int> postorderTraversal(TreeNode* root) {
		std::stack<TreeNode*> st;
		std::vector<int> vec;
		TreeNode* rt = root;

		while(rt || st.size())
		{
			while(rt)
			{
				st.push(rt->left);
				vec.push_back(rt->val);
				rt = rt->right;
			}
			rt = st.top();
			st.pop();
		}
		std::reverse(vec.begin(),vec.end());
		return vec;
	}
};

（2）
思路：每到一个节点 A，因为根要最后访问，将其入栈。然后遍历左子树，遍历右子树，最后返回到 A。但是出现一个问题，无法区分是从左子树返回，还是从右子树返回。因此，给 A 节点附加一个标记T。在访问其右子树前，T 置为 True。之后子树返回时，当 T 为 True表示从右子树返回，否则从左子树返回。当 T 为 false 时，表示 A 的左子树遍历完，还要访问右子树。同时，当 T 为 True 时，表示 A 的两棵子树都遍历过了，要访问 A 了。并且在 A 访问完后，A 这棵子树都访问完成了。

    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> S;
        unordered_map<TreeNode*,int> done;
        vector<int> v;
        TreeNode* rt = root;
        while(rt || S.size()){
            while(rt){
                S.push(rt);
                rt=rt->left;
            }
            while(S.size() && done[S.top()]){
                v.push_back(S.top()->val);
                S.pop();
            }
            if(S.size()){
                rt=S.top()->right;
                done[S.top()]=1;    
            }
        }
        return v;
    }