面试题:非递归前序遍历二叉树 &&非递归后序遍历

1,非递归前序遍历二叉树

基本思路:拿到一个节点先入vector,再看有没有右节点,有的话就入栈,否则不入,然后让节点向左走,循环判断

//Binary Tree Preorder Traversal
//非递归前序遍历二叉树



#include<stack>
#include<vector>
 struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
	
};

class Solution {
public:
	vector<int> preorderTraversal(TreeNode *root) {
		vector<int> v;
		if (root == NULL)
			return v;
		stack<TreeNode*> s;
		TreeNode* pCur = root;
		while (pCur || !s.empty())
		{
			if (pCur == NULL)
			{
				pCur=s.top();
				s.pop();
			}

			v.push_back(pCur->val);
			if (pCur->right)
				s.push(pCur->right);
			pCur = pCur->left;
		}
		return v;
	}
};

2.,非递归后序遍历

基本思路:

左边有就一直入栈,然后依次出栈,并查看 右边,右边不为空,那么就将右子树的左侧继续入栈。如果右侧为空 ,那么就直接入

vector。

注意:

有可能出现刚刚访问过一个右节点,这个右节点是叶子节点,那么接下来 会处理这个右节点 的父节点,就又会将这个右节点入栈,

所以要将 上一次处理过的节点进行保存并在处理 父节点时加以判定。

用图解释

//后序遍历二叉树
#include<stack>
#include<vector>
 struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
	
};

class Solution {
public:
	vector<int> postorderTraversal(TreeNode *root) {
		vector<int> v;
		if (root == NULL)
			return v;
		TreeNode*  pCur = root;
		TreeNode*  pPer = NULL;
		stack<TreeNode*> s;
		while (pCur || !s.empty())
		{
			//1,将根节点的左侧全部入栈
			while (pCur)
			{
				s.push(pCur);
				pCur = pCur->left;
			}
			//从栈中拿出一个分析,其左边肯定处理完了,检查右边就行
			pCur = s.top();
			if (pCur->right == NULL || pCur->right == pPer)//防止刚处理过右,又来将统一节点入栈,形成死循环
			{
				//右边为空,直接入vector
				s.pop();
				v.push_back(pCur->val);
				pPer = pCur;
				pCur = NULL;
			}
			else
			{
				//右边不为空,就让右的左侧全部入栈
				pCur = pCur->right;
			}
		}
	}
};

猜你喜欢

转载自blog.csdn.net/wm12345645/article/details/81712367