二叉树遍历非递归实现

二叉树遍历非递归实现

preOrder

下图是二叉树前序遍历的路径，可以发现遍历过程的路径总是沿着左边，直到当前节点左子树为空，再转向当前节点的右子树。

因此本文的主要逻辑如下：

1. 沿着左侧遍历，当遇到节点时访问节点并将当前节点入栈
2. 直到当前节点为空时，栈顶元素出栈（相当于返回上一状态），当前节点转向右子树
3. 重复执行第1、第2步

应当注意的是：什么时候我们认为二叉树的遍历完成呢？换句话说，就是我们第1、第2步循环的条件是什么？

本文认为这个条件就是，栈不为空，因为只要栈内还存在节点，即说明当前节点的父亲节点的右子树还未遍历过（当前节点的父亲节点的右子树可能为空）。

同时，因为初始状态时，栈本身就为空，要想进入第1、2步的循环内，本文将循环的条件完善为：

current != null || !stack.empty()

以确保程序能够进入循环。

public void preOrderNR() {
		Node current = root;
		Stack<Node> stack = new Stack<>();

		while (current != null || !stack.empty()) {
			// 一直向左，访问并压栈
			while (current != null) {
				System.out.println(current.e);
				stack.push(current);
				current = current.left;
			}
			// 出栈，转右
			current = stack.pop();
			current = current.right;
		}
	}

inOrder

主要逻辑如下：

1. 沿着左侧遍历，当遇到节点时将当前节点入栈
2. 直到当前节点为空时，栈顶元素出栈（相当于返回上一状态），访问节点，当前节点转向右子树
3. 重复执行第1、第2步

inOrder的非递归形式与preOrder的非递归形式是相似的，区别就在于访问节点的时刻不同。preOrder是在节点入栈时，访问节点；而inOrder则是在节点的出栈时访问节点。

	public void inOrderNR() {
		Node current = root;
		Stack<Node> stack = new Stack<>();

		while (current != null || !stack.empty()) {
			// 一直向左，并压栈
			while (current != null) {
				stack.push(current);
				current = current.left;
			}
			// 出栈，访问，并转右
			current = stack.pop();
			System.out.println(current.e);
			current = current.right;
		}
	}

leverOrder

对于层序遍历，本文借助队列实现。现在以下图讲解实现思路。

1. 首先让根节点A入队（A）
2. 队首节点A出队并访问，将A的左右子树B、C入队（B、C）
3. 队首节点B出队并访问，将B的左右子树D、E入队（C、D、E）
4. 队首节点C出队并访问，将C的左右子树入队（此时C左右子树都为空，不入队）（D、E）
5. 队首节点D出队并访问，将D的左右子树入队（此时D左右子树都为空，不入队）（E）
6. 队首节点E出队并访问，将E的左右子树入队（此时E左右子树都为空，不入队）（）

如此，我们访问节点的顺序便是节点出队的顺序：A、B、C、D、E

再进一步阐述逻辑：

1. 根节点入队
2. 队首节点出队并访问，将其左右非空子树入队
3. 重复第2步直到队列为空

public void leverOrder() {
		// 队列建立
		Queue<Node> queue = new LinkedList<>();
		queue.add(root);
		while (!queue.isEmpty()) {
			Node cur = queue.remove();
			System.out.println(cur.e);

			if (cur.left != null) {
				queue.add(cur.left);
			}
			if (cur.right != null) {
				queue.add(cur.right);
			}
		}
	}