马蜂窝2020校招算法方向（2019.9.23）第一道编程题

class Node:
    def __init__(self,data):
        self.data=data
        self.lchild=None
        self.rchild=None


class Tree:
    def __init__(self):
        self.queue=[]#应用队列存储树的节点
        self.flag=0#存储树根后flag置为1
        self.root=None

    #建树
    def createTree(self, list):
        while True:
            #list中没有数据，表现建树完成
            if len(list) == 0:
                return
            #flag为0，表现树根不存在
            if self.flag == 0:
                self.root = Node(list[0])
                #讲树根存入队列
                self.queue.append(self.root)
                #树根已创立，flag置为1
                self.flag = 1
                #剔除list中第一个已应用数
                list.pop(0)
            else:
                """
                treeNode:队列中的第一个节点(该节点左右孩子不完整存在)
                添加treeNode的左右孩子，当添加treeNode的右孩子以后，
                将队列中的第一个节点出队。
                """
                treeNode=self.queue[0]
                if treeNode.lchild==None:
                    treeNode.lchild=Node(list[0])
                    self.queue.append(treeNode.lchild)
                    list.pop(0)
                else:
                    treeNode.rchild = Node(list[0])
                    self.queue.append(treeNode.rchild)
                    list.pop(0)
                    self.queue.pop(0)

    # 队栈实现中序遍历
    def middle_queueAndStack(self, root):
        if root is None:
            return
        # 定义一个栈,存储节点
        stack = []
        node = root
        ans = []
        while stack or node:
            # 一直查找树的左节点,一直进栈
            while node:
                stack.append(node)
                node = node.lchild
            node = stack.pop()  # 该节点不存在左节点，该节点出栈，查找右节点
            ans.append(node.data)
            node = node.rchild

        return ans


if __name__ == "__main__":
    num = list(map(int, input().split(",")))
    point = eval(input())
    tree = Tree()
    tree.createTree(num)
    ans = tree.middle_queueAndStack(tree.root)
    output = -1
    for idx, i in enumerate(ans):
        if i == point and idx != len(ans) - 1:
            output = ans[idx + 1]
            break

    print(output)class Node:
    def __init__(self,data):