題目一：
從上到下按層打印二叉樹(shù)，同一層結(jié)點(diǎn)從左至右輸出。每一層輸出一行。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回二維列表[[1,2],[4,5]]
def Print(self, pRoot):
# write code here
#廣度優(yōu)先搜索（層次搜索）
#用?？梢院?jiǎn)單實(shí)現(xiàn)
if pRoot is None:
return []
result = []#存儲(chǔ)最終的輸出結(jié)果
currentstack = [pRoot] # 存儲(chǔ)當(dāng)前層入棧的節(jié)點(diǎn)
while currentstack:
res = [] #存儲(chǔ)該層的結(jié)果
nextstack = [] #存儲(chǔ)下一層入棧的節(jié)點(diǎn)
for i in currentstack:
res.append(i.val)
if i.left:
nextstack.append(i.left)
if i.right:
nextstack.append(i.right)
currentstack = nextstack
result.append(res)
return result
題目二：
請(qǐng)實(shí)現(xiàn)一個(gè)函數(shù)按照之字形打印二叉樹(shù)，即第一行按照從左到右的順序打印，第二層按照從右至左的順序打印，第三行按照從左到右的順序打印，其他行以此類(lèi)推。
思路：只需在上一題中新增一個(gè)time用來(lái)記錄當(dāng)前是奇數(shù)層還是偶數(shù)層即可；如果是偶數(shù)層，則將res中的元素反轉(zhuǎn)：res.reverse()或者是res = res[::-1]
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回二維列表[[1,2],[4,5]]
def Print(self, pRoot):
# write code here
#廣度優(yōu)先搜索（層次搜索）
#用?？梢院?jiǎn)單實(shí)現(xiàn)
if pRoot is None:
return []
result = []#存儲(chǔ)最終的輸出結(jié)果
currentstack = [pRoot] # 存儲(chǔ)當(dāng)前層入棧的節(jié)點(diǎn)
time = 0
while currentstack:
res = [] #存儲(chǔ)該層的結(jié)果
nextstack = [] #存儲(chǔ)下一層入棧的節(jié)點(diǎn)
time += 1
for i in currentstack:
res.append(i.val)
if i.left:
nextstack.append(i.left)
if i.right:
nextstack.append(i.right)
currentstack = nextstack
if time % 2== 0:
res.reverse()
result.append(res)
return result
