一、前序遍历
前序遍历可以通过一个stack,首先压入父节点,然后弹出父节点,输出其value,再压入右节点、左节点,至于为什么先压入右,是因为stack先入后出,会先弹出左节点,所以最终访问顺序是:自己 -> 左儿子 -> 右儿子
public void PreOrderUnrec(TreeNode root){
Stack<TreeNode> s=new Stack<>();
TreeNode p=root;
while(p!=null||!s.empty(){
while(p!=null){
System.out.println(p.val);
s.push(p);
p=p.left;
}
if(!s.empty()){
p=s.top();
s.pop();
p=p.right;
}
}
二、中序遍历
先把根节点压入栈,然后再将其所有左节点压入栈,然后弹出栈顶节点,输出其value,再将当前指针移到其右节点,若存在右节点,就下次循环可又把所有左节点压入栈中,这样保证访问顺序:左儿子 -> 自己 -> 右儿子
public void inOrderUnrec(TreeNode root){
Stack<TreeNode> s=new Stack<>();
TreeNode p=root;
while(p!=null||!s.empty(){
while(p!=null){
s.push(p);
p=p.left;
}
if(!s.empty()){
p=s.top();
s.pop();
System.out.println(p.val);
p=p.right;
}
}
三、后序遍历
//考察非递归的后序遍历
public class Solution {
public ArrayList<Integer> postorderTraversal(TreeNode root) {
ArrayList<Integer> list=new ArrayList<Integer>();
TreeNode p=root,r=null;//p记录当前节点,r记录上一次访问节点
Stack<TreeNode> s=new Stack<TreeNode>();
while(p!=null||!s.isEmpty()){
if(p!=null){//左孩子一直入栈,直到左孩子为空
s.push(p);
p=p.left;
}else{
p=s.peek();
p=p.right;
if(p!=null&&p!=r){//如果栈顶元素的左孩子不为空,且
s.push(p);
p=p.left;
}else{
p=s.pop();
list.add(p.val);
r=p;
p=null;
}
}
}
return list;
}
}