二叉树的先中后序遍历实现-非递归方式

还是拿上篇的二叉树举例：

一、先序遍历

步骤描述

需要借助一个栈，步骤如下：

1. 头节点压栈
2. 从栈顶弹出一个节点并打印，记为cur，如果栈为空，结束。
3. cur如有右孩子，压入右孩子。
4. cur如有左孩子，压入左孩子。
5. 重复步骤2，3，4。

举例详细说明

1. 头结点压栈
   
2. 从栈顶弹出一个节点并打印，此时栈为空
   

打印：
A

3. 栈中先压入右孩子，再压入左孩子
   
4. 弹出B，打印，并压入B的右孩子和左孩子
   

打印：
A B

5. 弹出C并打印，此时C没有左右孩子；
   

打印：
A B C

6. 继续弹出D并打印，此时D没有左右孩子;
   

打印：
A B C D

7. 继续弹出E并打印，先压入E的右孩子，再压入E的左孩子
   

打印：
A B C D E

8. 弹出F并打印，压入H
   

打印：
A B C D E F

9. 弹出H并打印，此时H没有左右孩子
   

打印：
A B C D E F H

10. 弹出G并打印， 此时G没有左右孩子
    

打印：
A B C D E F H G

12. 栈空，结束。

代码实现

Node节点定义：

public static class Node {
	public int value;
	public Node left;
	public Node right;

	public Node(int v) {
		value = v;
	}
}

先序遍历非递归方式：

public static void pre(Node head) {
	if (head != null) {
		Stack<Node> stack = new Stack<Node>();
		stack.add(head);
		while (!stack.isEmpty()) {
			head = stack.pop();
			System.out.print(head.value + " ");
			if (head.right != null) {
				stack.push(head.right);
			}
			if (head.left != null) {
				stack.push(head.left);
			}
		}
	}
	System.out.println();
}

二、中序遍历

分析：
一个二叉树，所有节点都能用如下的斜线完全覆盖，而这条斜线也就是某个节点和它所有的左子树的连线。

步骤描述

需要借助一个栈，步骤如下：

1. 记当前节点为cur，以cur为头的树的整条左边界进栈，直到遇到null停止；
2. 从栈中弹出一个节点cur并打印，把cur的右孩子记为cur，返回步骤1；
3. 如果栈为空，结束。

代码实现

public static void in(Node cur) {
	if (cur != null) {
		Stack<Node> stack = new Stack<Node>();
		while (!stack.isEmpty() || cur != null) {
			if (cur != null) {
				stack.push(cur);
				cur = cur.left;
			} else {
				cur = stack.pop();
				System.out.print(cur.value + " ");
				cur = cur.right;
			}
		}
	}
	System.out.println();
}

三、后序遍历

分析：

• 先序遍历时，先压右再压左，弹出时的顺序就是根左右，
• 如果先压左再压右，弹出顺序就是根右左，
• 如果我们此时先不打印，把弹出的节点按顺序压入另一个栈，待所有节点遍历完毕，再依次弹出另一个栈中的数据，此时弹出顺序就是左右根，也即是后序遍历的顺序。

步骤描述

需要借助2个栈（栈A，栈B），步骤如下：

1. 头结点压入栈A；
2. 从栈A弹出一个节点，记为cur，把弹出的节点cur压入栈B；如果栈A为空，结束。
3. cur如有左孩子，压入左孩子；
4. cur如有右孩子，压入右孩子；
5. 重复步骤2，3，4；
6. 依次弹出栈B中节点并打印，栈B为空，结束。

代码实现

public static void pos(Node head) {
	if (head != null) {
		Stack<Node> s1 = new Stack<Node>();
		Stack<Node> s2 = new Stack<Node>();
		s1.push(head);
		while (!s1.isEmpty()) {
			head = s1.pop(); 
			s2.push(head);
			if (head.left != null) {
				s1.push(head.left);
			}
			if (head.right != null) {
				s1.push(head.right);
			}
		}
		while (!s2.isEmpty()) {
			System.out.print(s2.pop().value + " ");
		}
	}
	System.out.println();
}

---

上一篇: 二叉树的先中后序遍历实现-递归方式