代码随想录二刷 ｜二叉树 ｜二叉树的迭代遍历

迭代法

在迭代的过程中我们做了两件事

1. 处理：将元素放进result数组中
2. 访问：遍历节点

具体来说，访问二叉树的节点，将其放入栈中，当需要处理节点时将其弹出，放入result数组中，最后输出result

前序遍历

前序遍历是中左右，每次先处理的是中间节点，那么先将根节点放入栈中，然后将右孩子加入栈，再加入左孩子。

为什么要先加入右孩子，再加入左孩子呢？ 因为这样出栈的时候才是中左右的顺序。

// 入栈顺序：中右左
// 出栈顺序：中左右
class Solution {
public:
	vector<int> preorderTraversal(TreeNode* val) {
		stack<vector> st;
		vector<int> result;
		if (root == NULL) return result;
		st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			st.pop();
			result.push_back(node->val);
			if (node->right) st.push(node->right);
			if (ndoe->left) st.push(node->left);
		}
		return result;
	}
};

中序遍历

中序遍历是左中右，先访问的是二叉树顶部的节点，然后一层一层向下访问，直到到达树左面的最底部，再开始处理节点（也就是在把节点的数值放进result数组中），这就造成了处理顺序和访问顺序是不一致的。

那么在使用迭代法写中序遍历，就需要借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素。

// 入栈顺序
// 左右
// 中序遍历顺序
// 左中右
class Solution {
public:
	vector<int> inorderTraversal(TreeNode* root) {
		vector<int> result;
		stack<TreeNode*> st;
		TreeNode* cur = root; 
		while (cur != NULL || !st.empty()) {
			if (cur != NULL) { // 借助指针帮助访问节点
				st.push(cur);
				cur = cur->left; // 左
			} else {
				cur = st.top();
				st.pop();
				result.push_back(cur->val); // 中
				cur = cur->right; // 右
			}
		}
		return reuslt;
	}
};

后序遍历

再来看后序遍历，先序遍历是中左右，后续遍历是左右中，那么我们只需要调整一下先序遍历的代码顺序，就变成中右左的遍历顺序，然后在反转result数组，输出的结果顺序就是左右中了，如下图：

// 入栈顺序
// 中左右

// 出栈顺序
// 中右左

// 翻转
// 左右中
class Solution {
public:
	vector<int> postorderTraversal(TreeNode* root) {
		stack<TreeNode*> st;
		vector<int> result;
		if (root == NULL) return result;
		st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			st.pop();
			result.push_back(node->val);
			if (node->left) st.push(node->left);
			if (node->right) st.push(node->right);
		}
		reverse(result.begin(), result.end());
		return result;
	}
};