二叉树的遍历

 

 

目录

一、深度遍历

1.1前序遍历

递归方法

非递归方法1

非递归方法2

1.2中序遍历

递归方法

非递归方法

1.3后序遍历

递归方法

非递归方法1

二、层次遍历

不分行

分行

之字形打印二叉树

代码

 

 


二叉树是一种非常重要的数据结构,很多其它数据结构都是基于二叉树的基础演变而来的。对于二叉树,有深度遍历和广度遍历,深度遍历有前序、中序以及后序三种遍历方法,广度遍历即我们平常所说的层次遍历。因为树的定义本身就是递归定义,因此采用递归的方法去实现树的三种遍历不仅容易理解而且代码很简洁,而对于广度遍历来说,需要其他数据结构的支撑。

      可利用栈将递归算法改写成非递归算法。

四种主要的遍历思想为:

1.深度遍历

  • 前序遍历:根结点 ---> 左子树 ---> 右子树
  • 中序遍历:左子树---> 根结点 ---> 右子树
  • 后序遍历:左子树 ---> 右子树 ---> 根结点

2.广度遍历(即层次遍历):只需按层次遍历即可

一、深度遍历

1.1前序遍历

递归方法

// 前序遍历-递归
	public static void preOrderRecur(TreeNode root) {
		if (root != null) {
			System.out.print(root.val + " ");
			preOrderRecur(root.left);
			preOrderRecur(root.right);
		}
	}

非递归方法1

        根据前序遍历的顺序,先访问根节点,然后再访问左子树和右子树。所以,对任意节点node,首先直接访问,之后再判断左子树是否为空,不为空时重复上述步骤,直至为空。若为空,则需要访问右子树。

       注意:在访问左子树之后,需要反过来访问其右子树,所以需要栈这种数据结构

步骤:

1.初始化一个空栈stack,申请一个节点空间pNode指向根节点(当前节点)

2.申请一个节点空间qNode,用来存放栈顶弹出的元素。

3.当p非空或者stack非空时,循环执行下面的操作

  • 如果p非空,访问该节点,并将节点p入栈,节点p再指向该节点的左孩子
  • 如果p为空,则弹出栈顶元素,将p指向该栈顶元素的右孩子。
// 前序遍历-非递归1
	public static void preOrderUnRecur1(TreeNode root) {
		Stack stack = new Stack<>();
		TreeNode pNode = root;
		while (pNode != null || !stack.isEmpty()) {
			if (pNode != null) {
				System.out.print(pNode.val + " ");// 访问根节点
				stack.push(pNode); // 根节点入栈
				pNode = pNode.left; // 遍历左子树
			} else {
				// pNode == null && !stack.isEmpty()
				TreeNode qNode = stack.pop(); // 出栈
				pNode = qNode.right; // 遍历右子树
			}

		}
	}

非递归方法2

这个方法比上个方法更容易理解一些。

1.初始化一个空栈stack,然后将头节点head压入栈中

2.从stack中弹出栈顶元素,记为cur,然后打印cur节点的值,再将cur的右孩子(不为空的话)压入栈中,最后将cur的左孩子压入栈中。

3.不断重复步骤2,直到stack为空,全部过程结束。

public static void preOrderUnRecur2(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack = new Stack<>();
		stack.push(root);
		while (!stack.isEmpty()) {
			TreeNode cur = stack.pop();
			System.out.print(cur.val + " ");
			// 注意:这里是将 右节点先入栈
			if (cur.right != null) {
				stack.push(cur.right);
			}
			if (cur.left != null) {
				stack.push(cur.left);
			}
		}
	}

1.2中序遍历

递归方法

// 中序遍历-递归
	public static void inOrderTraverse1(TreeNode root) {
		if (root != null) {
			inOrderTraverse1(root.left);
			System.out.print(root.val + " ");
			inOrderTraverse1(root.right);
		}
	}

非递归方法

与前序遍历类似,只是需要访问的顺序移到了出栈的时候

步骤:
     1.申请一个新的栈stack,初始时,令变量cur=root
     2.现将cur入栈,对于以cur为根节点的整棵子树来说,依次把左边界压入栈中,即不停的令cur = cur.left,然后重复步骤2
     3.不断重复步骤2直到发现cur为空,此时从stack中弹出一个节点,记为node,打印node的值,并且让cur = node.right,然后重复步骤2
 

// 中序遍历-非递归

	public static void inOrderUnRecur(TreeNode root) {
		if(root == null) {
			return;
		}
		Stack stack = new Stack<>();
		TreeNode cur = root;
		while(cur != null || !stack.isEmpty()) {
			if (cur != null) {
				stack.push(cur);
				cur = cur.left;
			}else {
				// 当cur为空时,即左边界走到底了
				TreeNode node = stack.pop();
				System.out.print(node.val + " ");
				cur = node.right;
			}
		}
		
	}

1.3后序遍历

递归方法

// 后序遍历-递归
	public static void postOrderTraverse1(TreeNode root) {
		if (root != null) {
			postOrderTraverse1(root.left);
			postOrderTraverse1(root.right);
			System.out.print(root.val + " ");
		}
	}

非递归方法1

后序遍历的非递归实现是三种遍历方式中最难的一种。因为在后序遍历中,要保证左孩子和右孩子都已被访问并且左孩子在右孩子前访问才能访问根结点,这就为流程的控制带来了难题。下面介绍两种思路。


     方法1:用两个栈实现后序遍历
     1.申请一个栈stack1,然后将头节点root入栈stack1
     2.从stack1中弹出来的节点记为cur,然后依次将cur的左孩子和右孩子入栈stack1
     3.在整个过程中,每一次从stack1中弹出来的节点都放进stack2中
     4.不断重复步骤2和3,直到stack1中为空,过程停止
     5.从stack2总依次弹出节点,打印的顺序就是后序遍历的顺序
   

// 后序遍历-非递归1
	public static void postOrderUnRecur1(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack1 = new Stack<>();
		Stack stack2 = new Stack<>();
		stack1.push(root);
		while(!stack1.isEmpty()) {
			TreeNode cur = stack1.pop();
			stack2.push(cur);
			if (cur.left != null) {
				stack1.push(cur.left);
			}
			if (cur.right != null) {
				stack1.push(cur.right);
			}
		}
		while(!stack2.isEmpty()) {
			System.out.print(stack2.pop().val + " ");
		}
	}

      方法2:要保证根结点在左孩子和右孩子访问之后才能访问,因此对于任一结点P,先将其入栈。如果P不存在左孩子和右孩子,则可以直接访问它;或者P存 在左孩子或者右孩子,但是其左孩子和右孩子都已被访问过了,则同样可以直接访问该结点。若非上述两种情况,则将P的右孩子和左孩子依次入栈,这样就保证了 每次取栈顶元素的时候,左孩子在右孩子前面被访问,左孩子和右孩子都在根结点前面被访问。

步骤:

1.初始化一个空栈stack,将头节点root入栈,同时设置两个申请变量cur和pre,在这个流程中cur代表stack的栈顶节点,pre代表最近一次弹出并打印的节点。初始时,cur为null,pre为头节点,

2.当stack非空时,循环执行下面的操作

  • 将栈顶节点置为当前节点cur,但是不从stack中弹出。
  • 如果当前节点没有左右孩子  或者 有左孩子或右孩子,但是已经被访问输出了:则访问该节点,并出栈,将其置为上一个访问的节点。
  • 如果不满足上面的条件的话,依次将右孩子和左孩子入栈。
// 后序遍历-非递归2
	public static void postOrderUnRecur2(TreeNode root) {
		if (root == null) {
			return ;
		}
		Stack stack = new Stack<>();
		stack.push(root);
		// 存放栈顶元素
		TreeNode cur = null;
		TreeNode pre = null;
		while(!stack.isEmpty()) {
			cur = stack.peek();
			//如果当前节点没有左右孩子  或者 有左孩子或右孩子,但是已经被访问输出了:则访问该节点,并出栈,将其置为上一个访问的节点。
			if ((cur.left == null && cur.right == null) ||
					(pre != null && (cur.left == pre || cur.right == pre))) {
				System.out.print(stack.pop().val + " ");
				pre = cur;
			}else {
				// 依次将cur的右孩子和左孩子入栈
				if (cur.right != null) {
					stack.push(cur.right);
				}
				if (cur.left != null) {
					stack.push(cur.left);
				}
			}
			
		}
		

二、层次遍历

不分行

二叉树的层序遍历的实现还是比较简单的,由于其层级的关系,很明显要用到队列来辅助实现,主要是从左向右,自上而下,依次将二叉树的各节点入队,这样便可以保证输出的顺序是层序排列的。下面是算法的实现思想:

1.初始化一个队列,首先将树的根节点入队

2.如果这个队列不为空,则进入循环

  • 将队首元素出队,并访问它
  • 如果该队首元素有左孩子,则将其左孩子入队
  • 如果该队首元素有右孩子,则将其左孩子入队
// 层次遍历
	public static void levelOrderTraverse1(TreeNode root) {
		if (root == null) {
			return;
		}
		// 用LinkenList来实现一个双向队列,这里我们看成是一个单项的。
		LinkedList queue = new LinkedList<>();
		queue.addLast(root);
		while(!queue.isEmpty()) {
			TreeNode pNode = queue.removeFirst();// 出队
			System.out.print(pNode.val + " ");
			if (pNode.left != null) {
				queue.addLast(pNode.left);// 入队
			}
			if (pNode.right != null) {
				queue.addLast(pNode.right);// 入队
			}
		}
		
	}

分行

从上到下按层打印二叉树,同一层的节点按从左往右的顺序打印,每一层打印到一行。

思路:和上面的类似,但是我们需要两个变量:

一个变量表示当前层中还没有打印的节点数;另一个变量表示下一层节点的数目;

	// 层次遍历:分行打印
		public static void levelOrderTraverse2(TreeNode root) {
			if (root == null) {
				return;
			}
			// 用LinkenList来实现一个双向队列,这里我们看成是一个单项的。
			// 分行打印,设置两个变量分别保存,该行剩余节点数和下一行节点数
			int nextLevel = 0;
			int toBePrint = 1;
			
			LinkedList queue = new LinkedList<>();
			queue.addLast(root);
			while(!queue.isEmpty()) {
				TreeNode pNode = queue.removeFirst();// 出队
				System.out.print(pNode.val + " ");
				toBePrint --;
				if (pNode.left != null) {
					queue.addLast(pNode.left);// 入队
					nextLevel ++;
				}
				if (pNode.right != null) {
					queue.addLast(pNode.right);// 入队
					nextLevel ++;
				}
				if (toBePrint == 0) {
					System.out.println();
					toBePrint = nextLevel;
					nextLevel = 0;
				}
			}
			
		}

 

之字形打印二叉树

题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。

// 层次遍历:之字形打印
	public static ArrayList> levelOrderTraverse3(TreeNode pRoot) {
		// 需要两个栈,A,B
		// 奇数层数字出栈后,先左后右保存在A
		// 偶数层数字出栈后,先右后左保存在B
		ArrayList> result = new ArrayList<>();
		if (pRoot == null) {
			return result;
		}
		Stack stackA = new Stack<>();
		Stack stackB = new Stack<>();

		stackB.push(pRoot);
		while (!stackA.isEmpty() || !stackB.isEmpty()) {
			ArrayList listB = new ArrayList<>();
			while (!stackB.isEmpty()) {
				TreeNode pNode = stackB.pop();

				listB.add(pNode.val);
//				System.out.print(pNode.val + " ");

				if (pNode.left != null) {
					stackA.push(pNode.left);
				}
				if (pNode.right != null) {
					stackA.push(pNode.right);
				}
			}
			if (!listB.isEmpty())
				result.add(listB);

			ArrayList listA = new ArrayList<>();
			while (!stackA.isEmpty()) {
				TreeNode pNode = stackA.pop();
				listA.add(pNode.val);
				if (pNode.right != null) {
					stackB.push(pNode.right);
				}
				if (pNode.left != null) {
					stackB.push(pNode.left);
				}

			}
			if (!listA.isEmpty()) {
				result.add(listA);
			}
		}

		return result;

	}
// 之字形打印方法2
	/*
	 * 思路:相较于[把二叉树打印成多行],该题的奇数行从左往右,偶数行从右往左。上题用的是队列,本题我们用双端队列
	 *
	 * 1.首先生成双端队列deque,将根节点从头部入队
	 * 2.如果是从左往右的过程:一律从头部弹出节点,如果有孩子,从尾部依次进入左孩子和右孩子:如第1,3,5,层
	 * 3.如果是从右往左的过程:一律从尾部弹出节点,如果有孩子,从头部依次进入右孩子和左孩子:如第2,4,6层
	 * 
	 */
	public static ArrayList> print(TreeNode pRoot) {
		ArrayList list = new ArrayList<>();
		ArrayList> result = new ArrayList<>();
		if (pRoot == null) {
			return result;
		}

		Deque deque = new LinkedList<>();
		deque.addFirst(pRoot);
		// flag 为true是奇数行
		boolean flag = true;
		// 本行剩余个数
		int count = 1;
		// 下一行个数
		int nextCount = 0;
		while (!deque.isEmpty()) {
			// 如果是奇数行,从左往右打印,一律从头部弹出节点,如果有孩子,从尾部依次进入左孩子和右孩子
			if (flag) {
				TreeNode node = deque.pollFirst();
				list.add(node.val);
				count--;
				if (node.left != null) {
					deque.addLast(node.left);
					nextCount++;
				}
				if (node.right != null) {
					deque.addLast(node.right);
					nextCount++;
				}
			} else {
				// 如果是偶数行,从右往左打印,一律从尾部弹出节点,如果有孩子,从头部依次进入右孩子和左孩子
				TreeNode node = deque.pollLast();
				list.add(node.val);
				count--;
				if (node.right != null) {
					deque.addFirst(node.right);
					nextCount++;
				}
				if (node.left != null) {
					deque.addFirst(node.left);
					nextCount++;
				}

			}

			if (count == 0) {
				result.add(list);
				list = new ArrayList<>();
				flag = !flag;
				count = nextCount;
				nextCount = 0;
			}
		}

		return result;

	}

代码

package basic;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Deque;
import java.util.LinkedList;
import java.util.Stack;
import basic.TreeNode;

/*
 public class TreeNode {
	public int val;
	public TreeNode left;
	public TreeNode right;
	public TreeNode(int x) {
		val = x;
	}

}
 */
public class TreeTest {

	// 前序遍历-递归
	public static void preOrderRecur(TreeNode root) {
		if (root != null) {
			System.out.print(root.val + " ");
			preOrderRecur(root.left);
			preOrderRecur(root.right);
		}
	}

	// 前序遍历-非递归1
	/*
	 * 1.初始化一个空栈stack,申请一个节点空间pNode指向根节点(当前节点) 2.申请一个节点空间qNode,用来存放栈顶弹出的元素。
	 * 3.当p非空或者stack非空时,循环执行下面的操作 如果p非空,访问该节点,并将节点p入栈,节点p再指向该节点的左孩子
	 * 如果p为空,则弹出栈顶元素,将p指向该栈顶元素的右孩子。
	 */
	public static void preOrderUnRecur1(TreeNode root) {
		Stack stack = new Stack<>();
		TreeNode pNode = root;
		while (pNode != null || !stack.isEmpty()) {
			if (pNode != null) {
				System.out.print(pNode.val + " ");// 访问根节点
				stack.push(pNode); // 根节点入栈
				pNode = pNode.left; // 遍历左子树
			} else {
				// pNode == null && !stack.isEmpty()
				TreeNode qNode = stack.pop(); // 出栈
				pNode = qNode.right; // 遍历右子树
			}

		}
	}

	// 前序遍历-非递归2:比非递归1更好理解
	/*
	 * 1.初始化一个空栈stack,然后将头节点head压入栈中
	 * 2.从stack中弹出栈顶元素,记为cur,然后打印cur节点的值,再将cur的右孩子(不为空的话)压入栈中,最后将cur的左孩子压入栈中。
	 * 3.不断重复步骤2,直到stack为空,全部过程结束
	 */
	public static void preOrderUnRecur2(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack = new Stack<>();
		stack.push(root);
		while (!stack.isEmpty()) {
			TreeNode cur = stack.pop();
			System.out.print(cur.val + " ");
			// 注意:这里是将 右节点先入栈
			if (cur.right != null) {
				stack.push(cur.right);
			}
			if (cur.left != null) {
				stack.push(cur.left);
			}
		}
	}

	// 中序遍历-递归

	public static void inOrderRecur(TreeNode root) {
		if (root != null) {
			inOrderRecur(root.left);
			System.out.print(root.val + " ");
			inOrderRecur(root.right);
		}
	}

	// 中序遍历-非递归
	/*
	 * 1.申请一个新的栈stack,初始时,令变量cur=root
	 * 2.现将cur入栈,对于以cur为根节点的整棵子树来说,依次把左边界压入栈中,即不停的令cur = cur.left,然后重复步骤2
	 * 3.不断重复步骤2直到发现cur为空,此时从stack中弹出一个节点,记为node,打印node的值,并且让cur =
	 * node.right,然后重复步骤2
	 */
	public static void inOrderUnRecur(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack = new Stack<>();
		TreeNode cur = root;
		while (cur != null || !stack.isEmpty()) {
			if (cur != null) {
				stack.push(cur);
				cur = cur.left;
			} else {
				// 当cur为空时,即左边界走到底了
				TreeNode node = stack.pop();
				System.out.print(node.val + " ");
				cur = node.right;
			}
		}

	}

	// 后序遍历-递归
	public static void postOrderRecur(TreeNode root) {
		if (root != null) {
			postOrderRecur(root.left);
			postOrderRecur(root.right);
			System.out.print(root.val + " ");
		}
	}

	/*
	 * 方法1:用两个栈实现后序遍历 1.申请一个栈stack1,然后将头节点root入栈stack1
	 * 2.从stack1中弹出来的节点记为cur,然后依次将cur的左孩子和右孩子入栈stack1
	 * 3.在整个过程中,每一次从stack1中弹出来的节点都放进stack2中 4.不断重复步骤2和3,直到stack1中为空,过程停止
	 * 5.从stack2总依次弹出节点,打印的顺序就是后序遍历的顺序
	 */
	// 后序遍历-非递归1
	public static void postOrderUnRecur1(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack1 = new Stack<>();
		Stack stack2 = new Stack<>();
		stack1.push(root);
		while (!stack1.isEmpty()) {
			TreeNode cur = stack1.pop();
			stack2.push(cur);
			if (cur.left != null) {
				stack1.push(cur.left);
			}
			if (cur.right != null) {
				stack1.push(cur.right);
			}
		}
		while (!stack2.isEmpty()) {
			System.out.print(stack2.pop().val + " ");
		}
	}

	/*
	 * 方法2:只用一个栈 1.初始化一个空栈stack,将头节点root入栈,同时设置两个申请变量cur和pre,
	 * 在这个流程中cur代表stack的栈顶节点,pre代表最近一次弹出并打印的节点。初始时,均为null。 2.当stack非空时,循环执行下面的操作
	 * 将栈顶节点置为当前节点cur,但是不从stack中弹出。 如果当前节点没有左右孩子  或者
	 * 有左孩子或右孩子,但是已经被访问输出了:则访问该节点,并出栈,将其置为上一个访问的节点。 如果不满足上面的条件的话,依次将右孩子和左孩子入栈。
	 */
	// 后序遍历-非递归2
	public static void postOrderUnRecur2(TreeNode root) {
		if (root == null) {
			return;
		}
		Stack stack = new Stack<>();
		stack.push(root);
		// 存放栈顶元素
		TreeNode cur = null;
		TreeNode pre = null;
		while (!stack.isEmpty()) {
			cur = stack.peek();
			// 如果当前节点没有左右孩子  或者 有左孩子或右孩子,但是已经被访问输出了:则访问该节点,并出栈,将其置为上一个访问的节点。
			if ((cur.left == null && cur.right == null) || (pre != null && (cur.left == pre || cur.right == pre))) {
				System.out.print(stack.pop().val + " ");
				pre = cur;
			} else {
				// 依次将cur的右孩子和左孩子入栈
				if (cur.right != null) {
					stack.push(cur.right);
				}
				if (cur.left != null) {
					stack.push(cur.left);
				}
			}

		}

	}

	// 层次遍历
	public static void levelOrderTraverse1(TreeNode root) {
		if (root == null) {
			return;
		}
		// 用LinkedList来实现一个双向队列,这里我们看成是一个单向的。

		LinkedList queue = new LinkedList<>();
		queue.addLast(root);
		while (!queue.isEmpty()) {
			// 出队
			TreeNode pNode = queue.removeFirst();
			System.out.print(pNode.val + " ");
			if (pNode.left != null) {
				// 入队
				queue.addLast(pNode.left);
			}
			if (pNode.right != null) {
				// 入队
				queue.addLast(pNode.right);
			}
		}

	}

	// 层次遍历:分行打印
	public static void levelOrderTraverse2(TreeNode root) {
		if (root == null) {
			return;
		}
		// 用LinkenList来实现一个双向队列,这里我们看成是一个单向的。
		// 分行打印,设置两个变量分别保存,该行剩余节点数和下一行节点数
		int nextLevel = 0;
		int toBePrint = 1;

		LinkedList queue = new LinkedList<>();
		queue.addLast(root);
		while (!queue.isEmpty()) {
			// 出队
			TreeNode pNode = queue.removeFirst();
			System.out.print(pNode.val + " ");
			toBePrint--;
			if (pNode.left != null) {
				// 入队
				queue.addLast(pNode.left);
				nextLevel++;
			}
			if (pNode.right != null) {
				// 入队
				queue.addLast(pNode.right);
				nextLevel++;
			}
			if (toBePrint == 0) {
				System.out.println();
				toBePrint = nextLevel;
				nextLevel = 0;
			}
		}

	}

	// 层次遍历:之字形打印
	public static ArrayList> levelOrderTraverse3(TreeNode pRoot) {
		// 需要两个栈,A,B
		// 奇数层数字出栈后,先左后右保存在A
		// 偶数层数字出栈后,先右后左保存在B
		ArrayList> result = new ArrayList<>();
		if (pRoot == null) {
			return result;
		}
		Stack stackA = new Stack<>();
		Stack stackB = new Stack<>();

		stackB.push(pRoot);
		while (!stackA.isEmpty() || !stackB.isEmpty()) {
			ArrayList listB = new ArrayList<>();
			while (!stackB.isEmpty()) {
				TreeNode pNode = stackB.pop();

				listB.add(pNode.val);
//				System.out.print(pNode.val + " ");

				if (pNode.left != null) {
					stackA.push(pNode.left);
				}
				if (pNode.right != null) {
					stackA.push(pNode.right);
				}
			}
			if (!listB.isEmpty())
				result.add(listB);

			ArrayList listA = new ArrayList<>();
			while (!stackA.isEmpty()) {
				TreeNode pNode = stackA.pop();
				listA.add(pNode.val);
				if (pNode.right != null) {
					stackB.push(pNode.right);
				}
				if (pNode.left != null) {
					stackB.push(pNode.left);
				}

			}
			if (!listA.isEmpty()) {
				result.add(listA);
			}
		}

		return result;
	}

	// 之字形打印方法2
	/*
	 * 思路:相较于[把二叉树打印成多行],该题的奇数行从左往右,偶数行从右往左。上题用的是队列,本题我们用双端队列
	 *
	 * 1.首先生成双端队列deque,将根节点从头部入队
	 * 2.如果是从左往右的过程:一律从头部弹出节点,如果有孩子,从尾部依次进入左孩子和右孩子:如第1,3,5,层
	 * 3.如果是从右往左的过程:一律从尾部弹出节点,如果有孩子,从头部依次进入右孩子和左孩子:如第2,4,6层
	 * 
	 */
	public static ArrayList> print(TreeNode pRoot) {
		ArrayList list = new ArrayList<>();
		ArrayList> result = new ArrayList<>();
		if (pRoot == null) {
			return result;
		}

		Deque deque = new LinkedList<>();
		deque.addFirst(pRoot);
		// flag 为true是奇数行
		boolean flag = true;
		// 本行剩余个数
		int count = 1;
		// 下一行个数
		int nextCount = 0;
		while (!deque.isEmpty()) {
			// 如果是奇数行,从左往右打印,一律从头部弹出节点,如果有孩子,从尾部依次进入左孩子和右孩子
			if (flag) {
				TreeNode node = deque.pollFirst();
				list.add(node.val);
				count--;
				if (node.left != null) {
					deque.addLast(node.left);
					nextCount++;
				}
				if (node.right != null) {
					deque.addLast(node.right);
					nextCount++;
				}
			} else {
				// 如果是偶数行,从右往左打印,一律从尾部弹出节点,如果有孩子,从头部依次进入右孩子和左孩子
				TreeNode node = deque.pollLast();
				list.add(node.val);
				count--;
				if (node.right != null) {
					deque.addFirst(node.right);
					nextCount++;
				}
				if (node.left != null) {
					deque.addFirst(node.left);
					nextCount++;
				}

			}

			if (count == 0) {
				result.add(list);
				list = new ArrayList<>();
				flag = !flag;
				count = nextCount;
				nextCount = 0;
			}
		}

		return result;

	}

	public static void main(String[] args) {
		TreeNode aNode = new TreeNode(1);
		TreeNode bNode = new TreeNode(2);
		TreeNode cNode = new TreeNode(3);
		TreeNode dNode = new TreeNode(4);
		TreeNode eNode = new TreeNode(5);
		TreeNode fNode = new TreeNode(6);
		TreeNode gNode = new TreeNode(7);
		TreeNode hNode = new TreeNode(8);

		aNode.left = bNode;
		aNode.right = cNode;
		bNode.left = dNode;
		bNode.right = eNode;
		cNode.right = fNode;
		eNode.left = gNode;
		eNode.right = hNode;
		System.out.print("前序遍历-递归 :   ");
		preOrderRecur(aNode);
		System.out.println();
		System.out.print("前序遍历-非递归1:");
		preOrderUnRecur1(aNode);
		System.out.println();
		System.out.print("前序遍历-非递归2:");
		preOrderUnRecur2(aNode);
		System.out.println();
		System.out.print("中序遍历-递归 :  ");
		inOrderRecur(aNode);
		System.out.println();
		System.out.print("中序遍历-非递归:");
		inOrderUnRecur(aNode);
		System.out.println();
		System.out.print("后序遍历-递归 :  ");
		postOrderRecur(aNode);
		System.out.println();
		System.out.print("后序遍历-非递归1:");
		postOrderUnRecur1(aNode);
		System.out.println();
		System.out.print("后序遍历-非递归2:");
		postOrderUnRecur2(aNode);
		System.out.println();
		System.out.print("层次遍历:");
		levelOrderTraverse1(aNode);
		System.out.println();
		System.out.println("层次遍历-分行打印:");
		levelOrderTraverse2(aNode);
		System.out.println("层次遍历-之字形打印1:");
		ArrayList> result = levelOrderTraverse3(aNode);

		for (ArrayList arrayList : result) {
			for (Integer val : arrayList) {
				System.out.print(val + " ");
			}
			System.out.println();
		}

		System.out.println("层次遍历-之字形打印2:");
		ArrayList> result1 = print(aNode);

		for (ArrayList arrayList : result1) {
			for (Integer val : arrayList) {
				System.out.print(val + " ");
			}
			System.out.println();
		}

	}
}

结果:

前序遍历-递归 :   1 2 4 5 7 8 3 6 
前序遍历-非递归1:1 2 4 5 7 8 3 6 
前序遍历-非递归2:1 2 4 5 7 8 3 6 
中序遍历-递归 :  4 2 7 5 8 1 3 6 
中序遍历-非递归:4 2 7 5 8 1 3 6 
后序遍历-递归 :  4 7 8 5 2 6 3 1 
后序遍历-非递归1:4 7 8 5 2 6 3 1 
后序遍历-非递归2:4 7 8 5 2 6 3 1 
层次遍历:1 2 3 4 5 6 7 8 
层次遍历-分行打印:
1 
2 3 
4 5 6 
7 8 
层次遍历-之字形打印1:
1 
3 2 
4 5 6 
8 7 
层次遍历-之字形打印2:
1 
3 2 
4 5 6 
8 7 

 

 

你可能感兴趣的:(java基础)