面试里必须搞懂----二叉树的前中后遍历

概述

二叉树在面试中占有不可或缺地位，所有要至少要搞明白前中后序的非递归写法，非递归遍历的原理是借助一个辅助栈。

中序遍历

中序遍历是左根右。

如果当前二叉树有左子树，那么就入栈。

如果没有左子数，那么先输出当前栈顶元素，然后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();
			}
		}
	}