我是一个初学python的人,正在尝试在codecademy上使用python和数据结构。
我正在学习链表。
我的作业是做一个方法来删除所有节点与给定的数据。我尝试了很多方法,但在删除2+节点AttributeError时一直卡住:'NoneType'对象没有属性'get_value'
我不知道我做错了什么。
class Node:
def __init__(self, value, next_node=None):
self.value = value
self.next_node = next_node
def get_value(self):
return self.value
def get_next_node(self):
return self.next_node
def set_next_node(self, next_node):
self.next_node = next_node
class LinkedList:
def __init__(self, value=None):
self.head_node = Node(value)
def get_head_node(self):
return self.head_node
def insert_beginning(self, new_value):
new_node = Node(new_value)
new_node.set_next_node(self.head_node)
self.head_node = new_node
def stringify_list(self):
string_list = ""
current_node = self.get_head_node()
while current_node:
if current_node.get_value() != None:
string_list += str(current_node.get_value()) + "n"
current_node = current_node.get_next_node()
return string_list
def remove_all_node(self, value_to_remove):
appeared_times = self.stringify_list().count(str(value_to_remove))
for i in range(0, appeared_times):
current_node = self.get_head_node()
if current_node.get_value() == value_to_remove:
self.head_node = current_node.get_next_node()
else:
while current_node:
next_node = current_node.get_next_node()
if next_node.get_value() == value_to_remove:
current_node.set_next_node(next_node.get_next_node())
current_node = None
else:
current_node = next_node
主要问题是:
-
self.stringify_list().count(str(value_to_remove))
不是正确的计数方法。例如,如果您有一个只有一个节点的树,其值为22,那么.count("2")
将返回2,因为它计数子字符串,因此这不是您需要的。 -
当
next_node
为None
时,if next_node.get_value() == value_to_remove
会报错。首先检查next_node
不是None
。
其他备注:
-
您的
LinkedList
构造函数不应该创建节点实例。它应该只将head_node
属性设置为None
,这是表示该列表具有没有节点的最准确方式。请注意,用户可能决定在列表中插入一个None
值,然后将其视为一个真正的节点。简而言之,不要在这个构造函数中创建节点。然后你也可以从stringify_list
方法中删除这个条件(当有一个真正的节点碰巧有None
作为值时,这会产生错误的效果):if current_node.get_value() != None
-
insert_beginning
的代码是正确的,但是您可以从Node
构造函数接受第二个参数的事实中获益。所以它可以是:def insert_beginning(self, new_value): self.head_node = Node(new_value, self.get_head_node())
-
在每次迭代中,您从列表的头部开始搜索一个事件。那是浪费时间。您应该在删除前一个出现的位置后继续搜索。
以下是remove_all_node
的更正版本:
def remove_all_node(self, value_to_remove):
current_node = self.get_head_node()
while current_node and current_node.get_value() == value_to_remove:
self.head_node = current_node = current_node.get_next_node()
while current_node:
next_node = current_node.get_next_node()
if next_node and next_node.get_value() == value_to_remove:
current_node.set_next_node(next_node.get_next_node())
else:
current_node = next_node