二叉树的迭代遍历-2

二叉树的递归遍历

二叉树的递归遍历方法较为简单，我在之前的博客中已经介绍过：

<二叉树的递归遍历>

递归的实现就是：每一次递归调用都会把函数的局部变量、参数值和返回地址等压入调用栈中，然后递归返回的时候，从栈顶弹出上一次递归的各项参数，因此递归可以返回上一层。

二叉树的迭代遍历：

二叉树也可以使用栈来实现迭代遍历，具体代码如下所示

1. 前序遍历是中左右，每次先处理的是中间节点，因此需要先将根节点放入栈中，然后将右子节点入栈，最后使左子节点入栈。因此前序遍历的代码最为简单
2. 但是如果使用上面的前序遍历的方式，中序遍历就无法实现了。因此前序遍历的访问顺序是中左右，先访问的是中间节点，要先处理的元素也是中间节点，因此可以很快写出简洁的前序遍历代码。
   但是中序遍历中的顺序为左中右，需要先访问方二叉树顶部的节点，然后一层一层向下访问，直到左子树的最底部，再开始处理节点，因此会导致处理顺序和访问顺序不一致。
   此时我们需要借助指针来指向访问节点，从而实现中序遍历。
3. 后序遍历为左右中，因此只需要把前序遍历稍微修改一下即可，先令其实现中左右的遍历，然后反转输出数组即可。

// 二叉树的前序遍历（迭代遍历） 
vector<int> preIterativeTraversal(TreeNode* root) {
	vector<int> vec;
	stack<TreeNode*> stk;
	
	if (root == NULL) return vec;
	
	// 将根节点插入到栈中
	stk.push(root); 
	
	while (!stk.empty()) {
		// 取出根节点
		TreeNode* temp = stk.top();
		stk.pop();
		
		vec.push_back(temp->val);
		
		// 将temp的右子节点插入
		if (temp->right) stk.push(temp->right);
		
		// 将temp的左子节点插入到stk中
		if (temp->left) stk.push(temp->left); 
	}
	
	return vec;
	
}

// 二叉树的中序遍历（迭代遍历） 
vector<int> inOrderIterativeTraversal(TreeNode* root) {
	// 二叉树的中序遍历比较复杂，需要使用指针指向当前节点
	vector<int> vec;
	stack<TreeNode*> stk;
	
	TreeNode* cur = root; 
	
	if (cur == NULL) return vec;
	
	while (cur != NULL || !stk.empty()) {
		// 如果当前节点不为空，则继续寻找左子节点，直到二叉树的末尾 
		if (cur != NULL) {
			stk.push(cur);
			cur = cur->left; 
		} else {
			// 如果左子节点为空，说明到达末尾，此时将左子节点存入数组中，并遍历右子节点
			cur = stk.top();
			stk.pop();
			
			vec.push_back(cur->val);
			
			// 将右子节点插入到stk中
			cur = cur->right; 
		}
	}
	
	return vec;
}

// 二叉树的后序遍历（迭代遍历） 
vector<int> postIterativeTraversal(TreeNode* root) {
	vector<int> vec;
	stack<TreeNode*> stk;
	
	if (root == NULL) return vec;
	
	// 将根节点插入到栈中
	stk.push(root); 
	
	while (!stk.empty()) {
		// 取出根节点
		TreeNode* temp = stk.top();
		stk.pop();
		
		vec.push_back(temp->val);
		
		// 将temp的左子节点插入
		if (temp->left) stk.push(temp->left); 
		
		// 将temp的右子节点插入到stk中
		if (temp->right) stk.push(temp->right);
		
	}
	
	// 后序遍历和前序遍历正好相反，当左右子节点插入顺序改变后，只需反转vec数组即可
	reverse(vec.begin(), vec.end()); 
	
	return vec;
}
} 

二叉树统一迭代遍历：

可以发现，当三种遍历方法不一致时，无法像递归遍历一样，简单的修改几行代码便可以实现三种遍历。但是统一遍历也是存在 一种方法的，具体的代码如下所示：

note: 虽然存在统一方法的迭代方法，但是该方法理解起来存在一定的难度。

// 二叉树的前序遍历（迭代遍历） 
vector<int> preIterativeTraversal(TreeNode* root) {
	// 二叉树的中序遍历比较复杂，需要使用指针指向当前节点
	vector<int> vec;
	stack<TreeNode*> stk;

	if (root) stk.push(root);
	
	while (!stk.empty()) {
		// 判断stk的头结点是否为空节点，如果为空节点表示中中间节点
		TreeNode *temp = stk.top();
		if (temp) {
			// 取出头节点，将左右子节点插入到栈中
			stk.pop();
			
			if (temp->right) stk.push(temp->right); // 右子节点
			if (temp->left) stk.push(temp->left); 	// 添加左子节点 
			stk.push(temp);							// 中节点访问过，但是还没有处理，添加空节点作为标记
			stk.push(NULL); 
			
		} else {
			// 如果当前的头结点为空节点，表示该节点之后的节点为中间节点
			// 1.首先把空节点弹出；2.然后弹出中间节点
			stk.pop();
			temp = stk.top();
			stk.pop();
			vec.push_back(temp->val); 
		}
	}
	
	return vec;
	
}

// 二叉树的中序遍历（迭代遍历） 
vector<int> inOrderIterativeTraversal(TreeNode* root) {
	// 二叉树的中序遍历比较复杂，需要使用指针指向当前节点
	vector<int> vec;
	stack<TreeNode*> stk;

	if (root) stk.push(root);
	
	while (!stk.empty()) {
		// 判断stk的头结点是否为空节点，如果为空节点表示中中间节点
		TreeNode *temp = stk.top();
		if (temp) {
			// 取出头节点，将左右子节点插入到栈中
			stk.pop();
			if (temp->right) stk.push(temp->right); // 右子节点
			stk.push(temp);							// 中节点访问过，但是还没有处理，添加空节点作为标记
			stk.push(NULL); 
			if (temp->left) stk.push(temp->left); 	// 添加左子节点 
		} else {
			// 如果当前的头结点为空节点，表示该节点之后的节点为中间节点
			// 1.首先把空节点弹出；2.然后弹出中间节点
			stk.pop();
			temp = stk.top();
			stk.pop();
			vec.push_back(temp->val); 
		}
	}
	return vec;
}

// 二叉树的后序遍历（迭代遍历） 
vector<int> postIterativeTraversal(TreeNode* root) {
	vector<int> vec;
	stack<TreeNode*> stk;

	if (root) stk.push(root);
	
	while (!stk.empty()) {
		// 判断stk的头结点是否为空节点，如果为空节点表示中中间节点
		TreeNode *temp = stk.top();
		if (temp) {
			// 取出头节点，将左右子节点插入到栈中
			stk.pop();
			stk.push(temp);							// 中节点访问过，但是还没有处理，添加空节点作为标记
			stk.push(NULL); 
			if (temp->right) stk.push(temp->right); // 右子节点
			if (temp->left) stk.push(temp->left); 	// 添加左子节点 
		} else {
			// 如果当前的头结点为空节点，表示该节点之后的节点为中间节点
			// 1.首先把空节点弹出；2.然后弹出中间节点
			stk.pop();
			temp = stk.top();
			stk.pop();
			vec.push_back(temp->val); 
		}
	}
	return vec;
}

参考文献：https://mp.weixin.qq.com/s/c_zCrGHIVlBjUH_hJtghCg