題目描述
請實現(xiàn)一個函數(shù)按照之字形打印二叉樹,即第一行按照從左到右的順序打印,第二層按照從右至左的順序打印,第三行按照從左到右的順序打印,其他行以此類推。
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
"""
由于需要打印Z字型,那么我們在遍歷整棵樹的時候就需要維護(hù)一個棧。
棧中存儲的是下一層的節(jié)點的逆序,則在訪問的時候按照棧的FILO特性就是正確的節(jié)點順序。
具體來說在打印奇數(shù)層的節(jié)點的時候,將該節(jié)點的【左右】子節(jié)點按順序入棧,
在打印偶數(shù)層的節(jié)點的時候,將該節(jié)點的【右左】節(jié)點按順序入棧。
總結(jié):遇到這類問題我們應(yīng)該先構(gòu)造一棵二叉樹然后按照題目要求模擬一遍,觀察出應(yīng)該使用什么樣的數(shù)據(jù)
結(jié)構(gòu),然后再進(jìn)行編程
"""
def Print(self, pRoot):
def helper(root_stack, ans, turns):
# 遞歸出口即隊列為空
if not root_stack:
return
length = len(root_stack)
temp = []
for i in range(length - 1, -1, -1):
# 對于某個節(jié)點,在打印完它的值之后,將其左右子節(jié)點先后入隊
temp.append(root_stack[i].val)
if root_stack[i].left:
root_stack.append(root_stack[i].left)
if root_stack[i].right:
root_stack.append(root_stack[i].right)
# 注意這里我們判斷當(dāng)前層的奇偶性之后再決定是【左右】還是【右左】
# 由于是對root_stack的最后兩個元素進(jìn)行位置交換,需要小心下標(biāo)越界問題
if turns & 1 != 0 and len(root_stack) > length + 1:
root_stack[-1], root_stack[-2] = \
root_stack[-2], root_stack[-1]
ans.append(temp)
helper(root_stack[length:], ans, turns + 1)
res = []
if not pRoot:
return res
helper([pRoot], res, 0)
return res
def main():
root = TreeNode(5)
root.left = TreeNode(4)
root.left.left = TreeNode(3)
root.left.left.left = TreeNode(2)
solution = Solution()
print(solution.Print(root))
if __name__ == '__main__':
main()
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)cdcxhl.cn,海內(nèi)外云服務(wù)器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應(yīng)用場景需求。