二叉树的迭代遍历
用压栈的方法来进行迭代 ,最关键的是用null标志来区分每个递归调用栈,不同遍历顺序只用改变压栈顺序即可(先序遍历:右节点->左结点->根节点 //中序遍历: 右节点->根节点->左结点 //后序遍历: 根节点->右节点->左结点)
我们在每次处理过的结点压栈后,在给他后面压一个空结点,这样栈顶元素如果是null就说明他下面的那个结点已经被处理过了,就直接访问就好了。
这样前序遍历、中序遍历、后序遍历就统一了!
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<Integer>();
Deque<TreeNode> stack = new LinkedList<TreeNode>();
//把根节点入栈
if(root!=null){
stack.push(root);
}
while(!stack.isEmpty()){
// 栈顶元素出栈
TreeNode cur = stack.pop();
if(cur!=null){
//先序遍历:右节点->左结点->根节点 //中序遍历: 右节点->根节点->左结点 //后序遍历: 根节点->右节点->左结点
//根节点
stack.push(cur);
stack.push(null);//打一个标记:标识是以这个节点为根节点的子树已经处理啦! // 右
if(cur.right!=null) stack.push(cur.right);
// 左
if(cur.left!=null) stack.push(cur.left);
}
else{ //当前栈顶元素为根节点的子树已经被处理的,所以就直接访问
//真正的访问
ans.add(stack.pop().val);
}
}
return ans;
}
}