编程语言
首页 > 编程语言> > 59.按之字形顺序打印二叉树(python)

59.按之字形顺序打印二叉树(python)

作者:互联网

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
 1 # -*- coding:utf-8 -*-
 2 # class TreeNode:
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.left = None
 6 #         self.right = None
 7 class Solution:
 8     def Print(self, pRoot):
 9         # write code here
10         if pRoot == None:
11             return []
12         stack1=[pRoot]
13         stack2=[]
14         ret=[]
15         while stack1 or stack2:
16             if stack1:
17                 tmpRet = []
18                 while stack1:
19                     tmpNode = stack1.pop()
20                     tmpRet.append(tmpNode.val)
21                     if tmpNode.left:
22                         stack2.append(tmpNode.left)
23                     if tmpNode.right:
24                         stack2.append(tmpNode.right)
25                 ret.append(tmpRet)
26             if stack2:
27                 tmpRet=[]
28                 while stack2:
29                     tmpNode = stack2.pop()
30                     tmpRet.append(tmpNode.val)
31                     if tmpNode.right:
32                         stack1.append(tmpNode.right)
33                     if tmpNode.left:
34                         stack1.append(tmpNode.left)
35                 ret.append(tmpRet)
36         return ret

2020-01-01 17:39:12

标签:之字形,59,stack2,tmpRet,right,二叉树,tmpNode,append,stack1
来源: https://www.cnblogs.com/NPC-assange/p/12129372.html