剑指offer系列-面试题32_1-分行从上到下打印二叉树(python)

1. 题目

分行从上到下打印二叉树。
从上到下按层打印二叉树， 同一层的节点按从左到右的顺序打印，每一层打印到一行。例如，打印图中的二叉树。

---

5

6

7

8

9

10

11

---

打印结果为：

8
6 10
5 7 9 11

2. 解题思路

这道题和剑指offer系列-面试题32-从上到下打印二叉树(python)是一样的思想只不过，最后打印出来的形式略有差异而已。

那么我们怎么知道队列中的节点是应该打印在第几行呢？或者说怎么知道到了哪一个节点，才需要换行呢？

我们需要两个变量:

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

3. 代码实现

def print_binarytree(root):
	"""
	:param root: 二叉树的根节点
	:return None
	"""
	if not root:
		return 
	queue = [] # 队列
	queue.append(root)
	to_be_print = 1 # 表示当前层中还没有打印的节点数
	next_level = 0 # 表示下一层节点的数目
	while queue:
		cur_node = queue.pop(0) # 时间复杂度为O(n)
		print(cur_node.value)
		if cur_node.left:
			queue.append(cur_node.left)
			next_level += 1 # 下一层有左子节点，下一层节点数目+1
		if cur_node.right:
			queue.append(cur_node.right)
			next_level += 1 # 下一层有右子节点，下一层节点数目+1
		# 每打印一个节点，to_be_print减1
		to_be_print -= 1
		# 如果to_be_print == 0表示这一层已经打印完毕，需要打印下一行了
		if to_be_print == 0:
			print('\n') # 换行
			to_be_print = next_level # 下一层需要打印的节点数变成了当前层中还没有打印的节点数
			next_levle = 0 # 同时清零下一层需要打印的节点数
	

4. 总结

这道题大部分是与面试题32是一样的，只是又增加了新的需求，使得逻辑复杂了一些。主要是还是要考虑清楚，怎么知道到哪个节点该换行，因此一定是需要变量来记录节点的一些信息，或者每行的信息，通过这些信息来判断是否该换行。

5. 参考文献

[1] 剑指offer丛书