Hackerrank Day 23: BST Level-Order Traversal 逐层遍历

最新推荐文章于 2020-01-12 20:58:05 发布

weixin_45405128

最新推荐文章于 2020-01-12 20:58:05 发布

阅读量158

点赞数

分类专栏：数据结构和算法 Python 编程笔试

本文链接：https://blog.csdn.net/weixin_45405128/article/details/103934697

版权

Python 同时被 3 个专栏收录

81 篇文章 1 订阅

订阅专栏

编程笔试

59 篇文章 0 订阅

订阅专栏

数据结构和算法

36 篇文章 0 订阅

订阅专栏

CODE

import sys

class Node:
    def __init__(self,data):
        self.right=self.left=None
        self.data = data
class Solution:
    def insert(self,root,data):
        if root==None:
            return Node(data)
        else:
            if data<=root.data:
                cur=self.insert(root.left,data)
                root.left=cur
            else:
                cur=self.insert(root.right,data)
                root.right=cur
        return root

    def levelOrder(self,root):
        #Write your code here
        nodes=[]#将节点存在list中
        results=''
        if root.data is not None:
            nodes.append(root)
        while nodes:
            tmp=nodes[0]
            results=results+str(tmp.data)+' '
            #print(str(tmp.data)+' ',end='')
            if tmp.left is not None:
                nodes.append(tmp.left)
            if tmp.right is not None:
                nodes.append(tmp.right)
            nodes.pop(0)#将对已经展开并存入result中的节点删除
        print(results)
T=int(input())
myTree=Solution()
root=None
for i in range(T):
    data=int(input())
    root=myTree.insert(root,data)
myTree.levelOrder(root)