**
二叉树的非递归遍历
**
中序:
中序遍历是先左孩子 后跟 最后右孩子的顺序。所以应该先一个while循环把所有的根及其左孩子依次入栈,然后一个while循环 将栈顶元素取出,遍历,如果有右孩子,同样需要将其及其左孩子入栈。代码如下
TreeNode tem = root;
LinkedList<TreeNode> stack = new LinkedList<>();
while(tem != null){
stack.push(tem);
tem = tem.left;
}
while(!stack.isEmpty()){
TreeNode node = stack.pop();
//假装遍历一下
System.out.print(node.val);
node = node.right;
while (node!=null){
stack.push(node);
node = node.left;
}
}
前序遍历:
前序遍历是先跟 然后左右孩子。所以直接将非null根结点入栈,然后一个while循环 将栈顶元素取出 遍历,如果有右孩子入队,有左孩子 入队,代码如下
if(root!=null){
LinkedList<TreeNode> stack = new LinkedList<>();
stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
//假装遍历
if(node.right != null){
stack.push(node.right);
}
if(node.left != null){
stack.push(node.left);
}
}
}
后序遍历:
后序遍历其实是相对较难的。但是也比较好理解
后序就是先左右孩子然后跟,所以我们任然需要将跟及其左孩子依次入栈,不同的是我们不能直接遍历,而是peek不是pop拿到到栈顶元素,如果该元素有右孩子并且未被访问过,那么我们就将右孩子入栈,并将其左孩子依次入栈。否则 访问该节点并更新刚刚访问过的结点到last 变量中(该变量的作用是判断是否被访问过的根据,如果一个元素
右孩子为刚刚访问过的,那么就不需要将右孩子及右孩子的左孩子依次入栈了)。代码如下
List<Integer> res = new ArrayList<>();
TreeNode last = null;
if(root!=null){
LinkedList<TreeNode> stack = new LinkedList<>();
while(root!=null){
stack.push(root);
root = root.left;
}
while (!stack.isEmpty()){
TreeNode node = stack.peek();
if(node.right != null && last != node.right){
TreeNode tem = node.right;
stack.push(tem);
while(tem.left!=null){
stack.push(tem.left);
tem = tem.left;
}
}else{
TreeNode node1 = stack.pop();
last = node1;
res.add(node1.val);
}
}
}
return res;