欢迎大家来到IT世界,在知识的湖畔探索吧!

#头条创作挑战赛#

前言

既然二叉树可以先序、中序、后序遍历，那应该也可以按照每层遍历，因此记录一下层序遍历，手写一边才能加深印象。

一、什么是层序遍历？

二叉树有高度，每一层高度上的从左到右的遍历就是层序遍历。
比如说我们有这样一颗二叉树：

我们想实现的遍历效果如下所示把每一层打印出来：
a
b c
d e f
k s t r
q z

二、思路

我不知道每层有几个节点、我也不知道树的高度
那么我就需要：

1. 记录每层的节点数
   1.1 上一层节点的左孩子+右孩子的个数，不就是我这层的节点总数嘛
   1.2 孩子可能没有，需要判断null值
2. 可以一层一层往下走
   2.1 每一层都需要记录下一层的数据，可以考虑队列
   2.2 队列当前层弹出做逻辑处理（打印），还可以将孩子加到下一层
3. 可以正常终止
   队列为空就终止了

三、实现

1. 建一颗树备用

代码如下：

		TreeNode head = new TreeNode("a");
		
		head.left = new TreeNode("b");
		head.right = new TreeNode("c");
		
		head.left.right = new TreeNode("d");
		head.right.left = new TreeNode("e");
		head.right.right = new TreeNode("f");
		
		head.left.right.left = new TreeNode("k");
		head.left.right.right = new TreeNode("s");
		head.right.left.right = new TreeNode("t");
		head.right.right.right = new TreeNode("r");
		
		head.right.left.right.right = new TreeNode("q");
		head.right.right.right.left = new TreeNode("z");

欢迎大家来到IT世界,在知识的湖畔探索吧!

2. 遍历逻辑

代码如下：

欢迎大家来到IT世界,在知识的湖畔探索吧!		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		queue.add(head);
		while (!queue.isEmpty()) {
			int size = queue.size();
			for (int i = 0; i < size; i++) {
				TreeNode nowNode = queue.poll();
				System.out.print(nowNode.val + "\t");
				if (nowNode.left != null) {
					queue.add(nowNode.left);
				}
				if (nowNode.right != null) {
					queue.add(nowNode.right);
				}
			}
			System.out.println();
		}

3. 实现效果

四、力扣原题

102. 二叉树的层序遍历
意思差不多，只不过我的是打印，这个是返回用二维数组封装，那就是将打印过程修改为add咯

	public List<List<Integer>> levelOrder(TreeNode root) {
		List<List<Integer>> ans = new LinkedList<>();
		if (root == null) {
			return ans;
		}
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		queue.add(root);
		while (!queue.isEmpty()) {
			int size = queue.size();
			List<Integer> nowAns = new LinkedList<>();
			for (int i = 0; i < size; i++) {
				TreeNode nowNode = queue.poll();
				nowAns.add(nowNode.val);
				if (nowNode.left != null) {
					queue.add(nowNode.left);
				}
				if (nowNode.right != null) {
					queue.add(nowNode.right);
				}
			}
			ans.add(nowAns);
		}
		return ans;
    }