題目描述
從上往下打印出二叉樹(shù)的每個(gè)節(jié)點(diǎn),同層節(jié)點(diǎn)從左至右打印。
創(chuàng)新互聯(lián)建站,為您提供重慶網(wǎng)站建設(shè)、重慶網(wǎng)站制作、網(wǎng)站營(yíng)銷推廣、網(wǎng)站開(kāi)發(fā)設(shè)計(jì),對(duì)服務(wù)餐廳設(shè)計(jì)等多個(gè)行業(yè)擁有豐富的網(wǎng)站建設(shè)及推廣經(jīng)驗(yàn)。創(chuàng)新互聯(lián)建站網(wǎng)站建設(shè)公司成立于2013年,提供專業(yè)網(wǎng)站制作報(bào)價(jià)服務(wù),我們深知市場(chǎng)的競(jìng)爭(zhēng)激烈,認(rèn)真對(duì)待每位客戶,為客戶提供賞心悅目的作品。 與客戶共同發(fā)展進(jìn)步,是我們永遠(yuǎn)的責(zé)任!
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
"""
由于需要逐層打印,那么我們?cè)诒闅v整棵樹(shù)的時(shí)候就需要維護(hù)一個(gè)隊(duì)列。
隊(duì)列中存儲(chǔ)的是下一層從左到右的節(jié)點(diǎn)。
具體來(lái)說(shuō)在打印第k層的節(jié)點(diǎn)的時(shí)候,將該節(jié)點(diǎn)的左右子節(jié)點(diǎn)按順序入隊(duì)即可。遞歸出口就是隊(duì)列為空
"""
def PrintFromTopToBottom(self, root):
def helper(root_queue, ans):
# 遞歸出口即隊(duì)列為空
if not root_queue:
return
length = len(root_queue)
for i in range(length):
# 對(duì)于某個(gè)節(jié)點(diǎn),在打印完它的值之后,將其左右子節(jié)點(diǎn)先后入隊(duì)
ans.append(root_queue[i].val)
if root_queue[i].left:
root_queue.append(root_queue[i].left)
if root_queue[i].right:
root_queue.append(root_queue[i].right)
helper(root_queue[length:], ans)
res = []
if not root:
return res
helper([root], res)
return res
def main():
root = TreeNode(8)
root.left = TreeNode(6)
root.right = TreeNode(10)
root.left.left = TreeNode(5)
root.left.right = TreeNode(7)
root.right.left = TreeNode(9)
root.right.right = TreeNode(11)
solution = Solution()
print(solution.PrintFromTopToBottom(root))
if __name__ == '__main__':
main()