从上往下打印二叉树 python_剑指offer-32从上到下打印二叉树-python

《剑指offer》python实现系列，全目录

第一想法：二叉树的广度优先遍历

1

2

3

4

5

6

7

8

9

10

11

12

13

14class Solution:

def PrintFromTopToBottom(self, root):

elemlist = []

if root is None:

return elemlist

queue = [root] #使用队列

while queue:

cur = queue.pop(0)

elemlist.append(cur.val)

if cur.left:

queue.append(cur.left)

if cur.right:

queue.append(cur.right)

return elemlist

升级版题目二：要求分行打印二叉树

​ 8 8

6 10 分行打印 ==》 6 10

5 7 9 11 5 7 9 11

方法：添加两个辅助变量，记录当前行待打印的节点数目toBePrintNum和下一行的节点个数nextLevelNum

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30class Solution:

# 返回二维列表[[1,2],[4,5]]

def Print(self, pRoot):

if pRoot is None:

return []

elemlist = []

levelElem = [] #一行的结点值

queue = [pRoot]#队列存储待处理结点

toBePrintNum = 1 #当前行待打印的节点数目

nextLevelNum = 0 #下一行节点数目

while queue:

curNode = queue.pop(0)

levelElem.append(curNode.val)

toBePrintNum -= 1

#若左存在，下一行节点数目+1，右孩子存在，下一行节点数目+1

if curNode.left:

queue.append(curNode.left)

nextLevelNum += 1

if curNode.right:

queue.append(curNode.right)

nextLevelNum += 1

if toBePrintNum == 0: #这一层的结点打印完了

elemlist.append(levelElem)

levelElem = [] #清空

toBePrintNum = nextLevelNum#更新为下一层节点数

nextLevelNum = 0

return elemlist

升级版题目三：之字形打印二叉树

第一行按从左到右，第二行按从右到左，第三行按从左到右

​ 8 8

6 10 分行打印 ==》 10 6

5 7 9 11 11 9 7 5

第一想法：在题目二的基础上，加一个reverseFlag，用于判断是否将levelElem进行reverse操作。

奇数行不reverse，偶数行reverse

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35class Solution:

def Print(self, pRoot):

if pRoot is None:

return []

reverseFlag = 1 #levlelElem[]是否需要reverse的flag,大于0表示不翻转

elemlist = []

levelElem = [] #一行的结点值

queue = [pRoot]#队列存储待处理结点

toBePrintNum = 1 #当前行待打印的节点数目

nextLevelNum = 0 #下一行节点数目

while queue:

curNode = queue.pop(0)

levelElem.append(curNode.val)

toBePrintNum -= 1

#若左存在，下一行节点数目+1，右孩子存在，下一行节点数目+1

if curNode.left:

queue.append(curNode.left)

nextLevelNum += 1

if curNode.right:

queue.append(curNode.right)

nextLevelNum += 1

if toBePrintNum == 0: #这一层的结点打印完了

if reverseFlag < 0:

levelElem.reverse()

reverseFlag *= (-1) #每处理完一行就更新

elemlist.append(levelElem)

levelElem = [] #清空

toBePrintNum = nextLevelNum#更新为下一层节点数

nextLevelNum = 0

return elemlist

剑指offer方法:感觉不如我的

​ 1

​ 2 3 【1.l 1.r 】

4 5 6 7 【3.r 3.l 2.r 2.l】

8 9 10 11 12 13 14 15 【4.l 4.r 5.l 5.r 6.l 6.r 7.l 7.r】

curNodeList记录当前层结点

利用栈的后进先出，添加一个辅助栈nextNodeStack，保存当前层每个结点的下一层的结点。

若待保存的是奇数层的子节点，先保存左孩再保存右孩子到nextStack栈里；

若打印的是偶数层的子节点，先保存右孩子再保存左孩子到nextStack栈里；

赋予curNodeList，nextStack清空，继续处理下一层。

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34class Solution:

def Print(self, pRoot):

if pRoot is None:

return []

isOddFlag = 1 #当前行是否是奇数行

curNodeList = [pRoot]

elemlist = []

while curNodeList:#每个while循环处理完当前行

nextNodeStack = [] #记录下一行的结点

levelElemlist = [] #存储当前行的结点值

if isOddFlag>0:#当前行是奇数行

for node in curNodeList:

levelElemlist.append(node.val)

#处理此行的子节点，先push左孩子再push右孩子

if node.left:

nextNodeStack.append(node.left)

if node.right:

nextNodeStack.append(node.right)

else:

for node in curNodeList:

levelElemlist.append(node.val)

#处理此行的子节点，先push右孩子再push左孩子

if node.right:

nextNodeStack.append(node.right)

if node.left:

nextNodeStack.append(node.left)

elemlist.append(levelElemlist)

levelElemlist = []

isOddFlag *= -1

nextNodeStack.reverse() #模拟栈先进后出，将list翻转就是先进后出了

curNodeList = nextNodeStack

return elemlist