马蜂窝 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)

先通过给定层次遍历序列得到二叉树，然后中序遍历二叉树，然后在中序遍历的结果中找当前节点的下一个即可。但是我只有91，应该有种什么情况没有考虑到，希望大佬们指点。

在牛客网上有大佬说当输入是非数字的情况下输出 -1, 我就很开(ri)心(gou)，好的，这种情况以后咱一定考虑到。