我正在学习python,并通过从头开始编写自己的链表来挑战自己。我使用的是一个内部节点类的传统结构,它包含一段数据和对下一个节点的引用。现在我正在尝试创建一个__repr__
方法,该方法返回节点的字符串表示。它返回的字符串如下所示:"This node contains {0}. The next node is {1}." .format(self.data, self.next.data)
除非列表中只有1个节点,否则它可以正常工作,这给了我以下错误:AttributeError: 'NoneType' object has no attribute 'data'.
为了解决这个问题,我首先检查列表中是否只有一个节点,然后返回以下字符串:"This node contains {0}. There is no next node." .format(self.data)
这就是我的__repr__
方法现在的样子:
def __repr__(self):
if MyLinkedList.get_size() == 1:
return "This node contains {0}. There is no next node." . format(self.data)
return "This node contains {0}. The next node is {1}." .format(self.data, self.next.data)
这就是目前为止整个链表类的样子:
class MyLinkedList(object):
head = None
size = None
def __init__(self):
self.size = 0
def get_head(self):
return self.head
def get_size(self):
return self.size
def is_empty(self):
if self.size == 0:
return True
else:
return False
def __repr__(self):
result = "["
curr = self.head
while curr != None:
if curr.next == None:
result += curr.data
break
result += curr.data
result += ", "
curr = curr.next
result += "]"
return result
def add_to_head(self, data):
new_node = MyLinkedList.Node(data)
if self.size == 0:
self.head = new_node
else:
new_node.next = self.head
self.head = new_node
self.size += 1
def delete_from_head(self):
if (self.size == 0):
self.head = None
else:
new_head = self.head.next
self.head = new_head
self.size =- 1
class Node():
next = None
def __init__(self, data):
self.data = data
def get_data(self):
return self.data
def get_next(self):
return self.next
def __repr__(self):
if MyLinkedList.get_size() == 1:
return "This node contains {0}. There is no next node." . format(self.data)
return "This node contains {0}. The next node is {1}." .format(self.data, self.next.data)
但现在,当我尝试打印任何节点的字符串表示时,它会给我以下错误:TypeError: get_size() missing 1 required positional argument: 'self'
有什么办法解决这个问题吗?我所要做的就是在内部节点类中调用外部类的get_size((方法,并检查该值是否为1。有没有其他方法可以使它在列表中只有一个节点时,使我的节点的__repr__
方法返回我希望它返回的字符串?
此外,如果您发现您可以对我的代码进行其他改进,我将很高兴听到这些改进。
您只能在MyLinkedList
类的实例上调用get_size()
。节点无论如何都不应该知道链表类的任何信息。只需利用节点的next
指针即可:
def __repr__(self):
suffix = "There is not next node" if self.next is None else "The next node is {}".format(self.next.data)
return "This node contains {}. {}.".format(self.data, suffix)
我认为您需要在get_size((方法中添加一个自参数。例如:
def __repr__(self):
if MyLinkedList.get_size(self) == 1:
return "This node contains {0}. There is no next node." . format(self.data)
return "This node contains {0}. The next node is {1}." .format(self.data, self.next.data)