前言:这个框架是模仿递归的前中后序来写的,应该算是比较好背的框架了
前序遍历
框架:
写这个框架的时候牢记以下:
-
cur是要遍历所有节点的指针,所有都遍历完之后cur会变长nullptr
-
前序遍历就按照递归的思路写即可。
-
res.push_back(cur->val) 等价于 递归中的操作
-
下面代码等价于preorder(root->left)
while(cur)
{
st.push(cur);
cur = cur->left;
}
- 下面代码等价于preorder(root->right)
TreeNode* top = st.top();
st.pop();
cur = top->right;
中序
框架:
和前序没什么差别,只是把操作拿到了中间。是不是和递归的改法一模一样呢
后序
框架
后序比前面两个要多维护一个lastNode。
这个变量是用来存储上一个节点访问的是谁
有这个变量的原因是:
代码执行来看,会不停的执行cur = top->right
因此要加多一个lastNode来记录,每当top->right = lastNode就代表右子树已经访问完了。
每当top->right != nullptr && top->right != lastNode就代表了右子树还没有访问完,需要继续访问,也就是cur = top->right;
当右子树访问完了,此时就进入了后序遍历的操作部分。
此时要做的工作
- 要更新一下lastNode
- 把值push_back进res
- 节点访问完了要从stack里面丢掉了