美文网首页
判断二叉树1

判断二叉树1

作者: dwq1666666 | 来源:发表于2019-12-24 18:27 被阅读0次
    """
    二叉搜索树定义:
        1,每个节点上都有数据
        2,当前节点的左节点数据比当前节点的数据小,并且当前节点的所有左节点数据都比当前节点的数据小
        3,当前节点的右节点数据比当前节点的数据大,并且当前节点的所有右节点数据都比当前节点的数据大
    """
    
    
    class Tree:
        def __init__(self):
            self.root = None
            self.list_data = []
    
        def insert(self, data):
    
            if self.root is None:
                self.root = Node(data)
                return
    
            current_node = self.root
    
            while current_node is not Node:
    
                if current_node.data == data:   # 这个数据已经在里面则跳出
                    return
    
                elif data > current_node.data:    # 当前这个节点的值小于需要插入的值
                    if current_node.right is None:      # 如果右节点为空插入这个节点数据
                        current_node.right = Node(data)
                    else:                               # 不为空的时候需要继续比较下一级
                        current_node = current_node.right
                else:   # 当前节点的数据大于需要插入的数据
                    if current_node.left is None:
                        current_node.left = Node(data)
                    else:
                        current_node = current_node.left
    
        def get_tree_data(self):
            self.print_node_data(self.root)
    
        def print_node_data(self, node):
                if node.left is not None:
                    self.print_node_data(node.left)
                self.list_data.append(node.data)
    
                if node.right is not None:
                    self.print_node_data(node.right)
    
        def print_tree(self, list_type=0):    # 遍历输出这颗树
            self.print_node(self.root, list_type)
    
        def print_node(self, node, list_type):
    
            if list_type == 0:  # 前序遍历
                if node.left is not None:
                    self.print_node(node.left, list_type)
                print(node.data)
    
                if node.right is not None:
                    self.print_node(node.right, list_type)
            elif list_type == 1:    # 前序遍历
                print(node.data)
                if node.left is not None:
                    self.print_node(node.left, list_type)
                if node.right is not None:
                    self.print_node(node.right, list_type)
            elif list_type == 2:    # 后序遍历
                if node.left is not None:
                    self.print_node(node.left, list_type)
                if node.right is not None:
                    self.print_node(node.right, list_type)
                print(node.data)
    
        # 判断是否是一颗二叉树
        # def is_tree(self):
        #     return self.is_tree_detail(self.root)
        #
        # def is_tree_detail(self, node):
        #     flag = True
        #     if node is None:
        #         return flag
        #
        #     # 处理当前节点
        #     if self.is_fit(node.data, node) is False:
        #         return False
        #
        #     # 处理当前节点的左节点
        #     if self.is_tree_detail(node.left) is False:
        #         return False
        #
        #     # 处理当前节点的右节点
        #     if self.is_tree_detail(node.right) is False:
        #         return False
        #
        #     return flag
        #
        # def is_fit(self, data, node):
        #     flag = True
        #
        #     if node.left is not None:
        #         if data < node.left.data:
        #             return False
        #         else:
        #             if self.is_fit(data, node.left) is False:
        #                 return False
        #
        #     if node.right is not None:
        #         if data > node.right.data:
        #             return False
        #         else:
        #             if self.is_fit(data, node.right) is False:
        #                 return False
        #
        #     return flag
    
    
    class Node:
        def __init__(self, data):
            self.data = data
            self.left = None
            self.right = None
    
    
    def get_node(data, node_id):
    
        if node_id == 0:
            return None
    
        for x in data:
            if x[0] == node_id:
                # print('get_node:node_id:{}'.format(node_id), x[1])
                node = Node(x[1])
                node.left = get_node(data, x[2])
                node.right = get_node(data, x[3])
                return node
    
        return None
    
    
    def main():
        n = int(input())
        data = [[int(i) for i in input().split()] for j in range(n)]    # 一次性读入所有的数据
    
        # print(data)
    
        # 存储二叉树
        tree = Tree()
        for i in range(n):
            x = data[i]
            if x[0] == 1:
                node = Node(x[1])
                node.left = get_node(data, x[2])
                node.right = get_node(data, x[3])
                tree.root = node
    
        # tree.print_tree()
        tree.get_tree_data()
        list_data_new = tree.list_data[:]
        list_data_new.sort()
    
        # 如果是一颗二叉树的,那么得到的数据和排序后输出的数据肯定是一致的
        flag = True
        for i in range(len(list_data_new)):
            if tree.list_data[i] != list_data_new[i]:
                flag = False
    
        if flag:
            print('T')
        else:
            print('F')
    
        return
    
    
    if __name__ == '__main__':
        main()
    
    

    相关文章

      网友评论

          本文标题:判断二叉树1

          本文链接:https://www.haomeiwen.com/subject/nhvfoctx.html