二叉树的非递归遍历怎么写？

Posted 2021-07-18 _BitterSweet

非递归实现二叉树的遍历

• 二叉树的前序遍历
• • 题目：力扣144.二叉树的前序遍历
  • 代码实现：
• 二叉树的中序遍历
• • 题目：力扣94.二叉树的中序遍历
  • 代码实现：
• 二叉树的后序遍历
• • 题目：力扣145.二叉树的后序遍历
  • 代码实现：

二叉树的前序遍历

• 前序遍历是根-->左-->右，每次先处理的是中间节点，那么先将跟节点放⼊栈中，然后将右孩⼦加⼊栈，再加⼊左孩⼦
• 为什么要先加入右孩子呢，因为出栈的顺序是先进后出，这样才能实现根--> 左--> 右的出栈顺序
  

题目：力扣144.二叉树的前序遍历

代码实现：

class Solution 
{
public:
	vector<int> preorderTraversal(TreeNode* root) 
	{
		stack<TreeNode*> st;
		vector<int> ans;
		if (root == NULL) return ans;
		st.push(root);
		while (!st.empty()) 
		{
			TreeNode* node = st.top(); // 根
			st.pop();
			ans.push_back(node->val);
			if (node->right) st.push(node->right); // 右（空节点不⼊栈）
			if (node->left) st.push(node->left); // 左（空节点不⼊栈）
		}
		return ans;
	}
};

二叉树的中序遍历

• 中序遍历的顺序是左--> 根-->右，先访问的是二叉树顶部的节点，然后一层一层向下访问，直到左树的最底部，再处理节点，
• 我们需要定义一个指针来访问节点，用栈来处理访问的元素
  

题目：力扣94.二叉树的中序遍历

代码实现：

class Solution 
{
public:
	vector<int> inorderTraversal(TreeNode* root) 
	{
		vector<int> ans;
		stack<TreeNode*> st;
		TreeNode* cur = root;
		while (cur != NULL || !st.empty()) 
		{
			if (cur != NULL) 
			{ // 指针来访问节点，访问到最底层
				st.push(cur); // 将访问的节点放进栈
				cur = cur->left; // 左
			}
			else 
			{
				cur = st.top(); // 从栈⾥弹出的数据，就是要处理的数据（放进ans数组⾥的数据）
					st.pop();
				ans.push_back(cur->val); // 根
				cur = cur->right; // 右
			}
		}
		return ans;
	}
};

二叉树的后序遍历

• 后序遍历的顺序是：左-->右-->根，和前序相比，我们会发现，我们可以先调整代码的左右顺序，变成根 -->右-->左，再反转vector数组，及变成了我们的左-->右-->根
  

题目：力扣145.二叉树的后序遍历

代码实现：

class Solution 
{
public:
	vector<int> postorderTraversal(TreeNode* root) 
	{
		stack<TreeNode*> st;
		vector<int> ans;
		if (root == NULL) return ans;
		st.push(root);
		while (!st.empty()) 
		{
			TreeNode* node = st.top();
			st.pop();
			ans.push_back(node->val);
			if (node->left) st.push(node->left); // 相对于前序遍历，这更改⼀下⼊栈顺序 （空节点不⼊栈）
				if (node->right) st.push(node->right); // 空节点不⼊栈
		}
		reverse(ans.begin(), ans.end()); // 将结果反转之后就是左右中的顺序了
		return ans;
	}
};