概述
二叉树在面试中占有不可或缺地位,所有要至少要搞明白前中后序的非递归写法,非递归遍历的原理是借助一个辅助栈。
中序遍历
中序遍历是左根右。
如果当前二叉树有左子树,那么就入栈。
如果没有左子数,那么先输出当前栈顶元素,然后p在走到右子数,不断的重复这个过程,说实话这个代码不好想
public void midOrder(TreeNode root){
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode p = root;
while(p!=null || !stack.isEmpty()){
while(p!=null){
stack.push(p);
p = p.left;
}
p = stack.pop();
System.out.print(p.val+" ");
p = p.right;
}
}
前序遍历
非递归的前序遍历和中序遍历是很像的,只要改一个地方,每次先输出当前节点元素即可。
public void preOrder(TreeNode root){
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode p = root;
while(p!=null || !stack.isEmpty()){
while(p!=null){
System.out.print(p.val+" ");
stack.push(p);
p = p.left;
}
p = stack.pop().right;
}
}
后续遍历
后序遍历是比较难的,左右根,是先添加左儿子,在添加右儿子,在回溯的时候添加根节点,我们并不知道什么时候右子数已经遍历完。
所以一般有经验的面试管不会考后序遍历的非递归写法
存在一个取巧的办法:
前序遍历是 根左右
后序遍历是 左右根
将后续遍历翻转正好是 根右左,可以用前序遍历实现。
public void postOrder(TreeNode root){
TreeNode curr = root, pre = null;
Stack<TreeNode> stack = new Stack<TreeNode>();
while(curr!=null || !stack.isEmpty()){
while(curr!=null){
stack.push(curr);
curr = curr.left;
}
TreeNode top = stack.peek();
if(top.right!=null && pre!=top.right){
curr = top.right;
}
else{
// pre指向出栈节点,以便于后面判断是否为栈顶节点的右孩子
// 如果是,那么栈顶元素继续出栈,否则栈顶元素右孩子入栈
pre = top;
System.out.print(top.val+" ");
stack.pop();
}
}
}