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;
}
}
}
};