[LeetCode]Binary Tree Level Order Traversal

最新推荐文章于 2021-11-17 20:57:12 发布

原创最新推荐文章于 2021-11-17 20:57:12 发布 · 497 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

102 篇文章

订阅专栏

本文详细介绍了如何通过队列实现二叉树的层次遍历，并提供了具体的代码实现。通过逐层添加节点到队列中，然后从队列中依次取出节点并打印其值，最终得到按照层次顺序输出的节点值列表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree {3,9,20,#,#,15,7},

return its level order traversal as:

[
  [3],
  [9,20],
  [15,7]
]

按层输出二叉树。

解题思路

采用队列，逐层添加到队列中，每次将队首的元素取出并将其孩子节点添加到队尾，直到队列为空为止。

代码

        /**
	 * 按层输出二叉树
	 * @param root
	 * @return
	 */
	public static List<List<Integer>> levelOrder(TreeNode root) {
		List<List<Integer>> levelList = new ArrayList<List<Integer>>();
		if (root == null)
			return levelList;
		
		Queue<TreeNode> queue = new LinkedList<TreeNode>();//队列
		TreeNode temp;
		int lastLevelLen = 1;//上一层元素个数
		int currentLevelLen = 0;//当前层元素个数
		List<Integer> list = new ArrayList<Integer>();

		queue.add(root);//添加根节点到队列
		while (!queue.isEmpty()) {
			temp = queue.poll();

			list.add(temp.val);
			lastLevelLen--;

			if (temp.left != null) {
				//如果左孩子不为空，将其添加到队尾，当前层节点个数加一
				queue.offer(temp.left);
				currentLevelLen++;
			}

			if (temp.right != null) {
				//如果右孩子不为空，将其添加到队尾，当前层节点个数加一
				queue.offer(temp.right);
				currentLevelLen++;
			}

			if (lastLevelLen == 0) {
				//如果上一层已经遍历完毕，且已将上一层的孩子节点添加到队列里
				lastLevelLen = currentLevelLen;
				currentLevelLen = 0;
				levelList.add(list);
				list =  new ArrayList<Integer>();
			}
		}
		return levelList;
	}