用迭代法统一风格实现二叉树的遍历

前言

前一篇文章学习完后，我发现风格不是很统一，除了先序和后序有关联，中序完全就是另一个风格了，一会用栈遍历，一会又用指针来遍历。那么，如何实现“一招鲜，吃遍天”的统一写法呢？

思路

这里以中序遍历为例，还是用栈的解决思路。将访问的节点放入栈中， 把要处理的节点也放入栈中，但是要做标记。
中序遍历动画如下：

以上用例的中序遍历是：1、4、2、5、6 （左、中、右）
动画中，result数组就是最终结果集。
可以看出，我们将访问的节点，直接加入栈中。 但如果是处理的节点，则后面放入一个空节点，这样只有空节点弹出的时候，才将下一个节点放进结果集。

代码

中序遍历：

	class Solution {
		public:
		    vector<int> inorderTraversal(TreeNode* root) {
		        vector<int> result;
		        stack<TreeNode*> st;
		        if (root != NULL) st.push(root);
		        while (!st.empty()) {
		            TreeNode* node = st.top();
		            if (node != NULL) {
		                st.pop(); // 将该节点弹出，避免重复操作，下面再将右中左节点添加到栈中
		                if (node->right) st.push(node->right);  // 添加右节点（空节点不入栈）
		
		                st.push(node); // 添加中节点
		                st.push(NULL); // 中节点访问过，但是还没有处理，加入空节点做为标记。
		
		                if (node->left) st.push(node->left); // 添加左节点（空节点不入栈）
		            } else { // 只有遇到空节点的时候，才将下一个节点放进结果集
		                st.pop(); / 将空节点弹出
		                node = st.top(); / 重新取出栈中元素
		                st.pop();//将栈顶清空
		                result.push_back(node->val); // 加入到结果集
		            }
		        }
		        return result;
		    }
	};

前序遍历（仅仅改变了两行代码）：

	            if (node != NULL) {
	                st.pop();
	                if (node->right) st.push(node->right);  // 右
	                if (node->left) st.push(node->left);    // 左
	                st.push(node);                          // 中
	                st.push(NULL);
	            } 

后序遍历（仅仅改变了两行代码）：

if (node != NULL) {
                st.pop();
                st.push(node);                          // 中
                st.push(NULL);
                if (node->right) st.push(node->right);  // 右
                if (node->left) st.push(node->left);    // 左
            }