Почему мой параметр / объект отображается как объект NoneType? - PullRequest
1 голос
/ 08 октября 2019

Я учу себя бинарному дереву поиска, и в этой программе я вставляю данные в дерево, но возникает ошибка 'NoneType' object has no attribute 'data'.

from collections import deque
class Node:
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

def insert(rnode, data):
    if data > rnode.data:
        insert(rnode.right, data)
        rnode.right.data = data
        return print('Node inserted')
    if data < rnode.data:
        insert(rnode.left, data)
        rnode.left.data = data
        return print('Node inserted')

def bfs():
    q = deque()
    temp = self.root
    while temp:
        print(temp.data)
        q.append(temp.left)
        q.append(temp.right)
        temp = q.popleft()


root = Node(8)
insert(root, 3)
bfs()

ОШИБКА:

Exception has occurred: AttributeError
'NoneType' object has no attribute 'data'
  File "/home/mayank/Documents/datastructures/binarysearchtree.py", line 9, in insert
    if data > rnode.data:
  File "/home/mayank/Documents/datastructures/binarysearchtree.py", line 14, in insert
    insert(rnode.left, data)
  File "/home/mayank/Documents/datastructures/binarysearchtree.py", line 29, in <module>
    insert(root, 3)

Помощь !!

1 Ответ

0 голосов
/ 08 октября 2019

Ваша функция insert предполагает, что rnode.right и rnode.left всегда ссылаются на дерево, в которое можно вставить значение, но это не тот случай, если rnode является листом (который, в конечном итоге, будетпри вставке нового значения) insert необходимо создать новый узел для нового значения.

def insert(rnode, data):
    if data > rnode.data:
        if rnode.right is None:
            rnode.right = Node(data)
        else:
            insert(rnode.right, data)
    if data < rnode.data:
        if rnode.left is None:
            rnode.left = Node(data)
        else:
            insert(rnode.left, data)
...