声明:本文是笔者在观看b站视频后学习的一个的记录,仅作为个人复习用,如有侵权请联系笔者删除,视频链接:一个模板解决二叉树前中后遍历的迭代写法!
递归的底层实现实际上是利用系统的程序栈,于是我们可以通过自己建立一个栈模拟程序栈的方法来将二叉树的递归遍历改写成迭代算法,这需要我们深刻理解二叉树的递归遍历过程和前中后序访问的时机。
我们知道二叉树在进行递归时总是先递归左子树一直到叶节点,然后再递归右子树一直到叶节点,最后回到根节点。前序遍历的位置是在左右子树还未开始遍历时;中序遍历的位置是左子树已经被遍历完而右子树还未开始遍历;后序遍历的位置是左右子树都遍历完了。
于是在模拟程序栈遍历二叉树时,我们可以采用以下思路:先不断把根节点和左孩子压入栈中,然后出栈时看看这个节点是否有右孩子,有的话把右孩子压入栈中,再把右孩子的左孩子节点不断压入栈中,重复这样的操作知道栈为空。
下面是前中后序具体代码:
前序:
前序遍历的位置在节点压入栈之前
class Solution{
public List<Integer> traverse(TreeNode root){
List<Integer> ans = new ArrayDeque<>(); //存放遍历结果
Stack<TreeNode> stack = new ArrayDeque<>(); //模拟函数栈
while(root!=null){ //先把左侧树枝一撸到底
//前序遍历位置
ans.add(root.val);
stack.push(root);
root = root.left;
}
while(!stack.isEmpty()){
TreeNode p = stack.pop();
p = p.right; //弹出之后看是否有右孩子
while(p!=null){ //有右孩子的话把右孩子的左侧树枝一撸到底
//前序遍历位置
ans.add(p.val);
stack.push(p);
p = p.left;
}
}
}
}
中序:
中序遍历的位置在节点从栈中pop出来之后
class Solution{
public List<Integer> traverse(TreeNode root){
List<Integer> ans = new ArrayDeque<>(); //存放遍历结果
Stack<TreeNode> stack = new ArrayDeque<>(); //模拟函数栈
while(root!=null){ //先把左侧树枝一撸到底
stack.push(root);
root = root.left;
}
while(!stack.isEmpty()){
TreeNode p = stack.pop();
//中序遍历位置
ans.add(p.val);
p = p.right; //弹出之后看是否有右孩子
while(p!=null){ //有右孩子的话把右孩子的左侧树枝一撸到底
stack.push(p);
p = p.left;
}
}
}
}
后序:
后序这里需要用到一个技巧,我们知道后序遍历的顺序是左右中,前序遍历的顺序是中左右,那么如果我们把前序遍历的代码稍微更改,使其遍历顺序变成中右左(这个是好改的,因为左右是对称的,只需要把left,right颠倒即可),然后在向ans中添加答案时采用头插法,这样遍历结果的顺序就又颠倒了,变成了左右中
class Solution{
public List<Integer> traverse(TreeNode root){
List<Integer> ans = new ArrayDeque<>(); //存放遍历结果
Stack<TreeNode> stack = new ArrayDeque<>(); //模拟函数栈
while(root!=null){ //先把右侧树枝一撸到底
//后序遍历位置
ans.add(0, root.val);
stack.push(root);
root = root.right;
}
while(!stack.isEmpty()){
TreeNode p = stack.pop();
ans.add(p.val);
p = p.left; //弹出之后看是否有左孩子
while(p!=null){ //有左孩子的话把左孩子的右侧树枝一撸到底
//后序遍历位置
ans.add(0, p.val);
stack.push(p);
p = p.right;
}
}
}
}