我正在编写链表的实现,链表类及其节点如下所示:
class Node:
def __init__(self, data, next = None):
self.element = data
self.next = next
class LinkedList:
def __init__(self, head):
self.head = head
def insert(self, next):
node = self.head
while node.next:
node = node.next
node.next = next
def remove(self, element):
node = self.head
while node:
if node.element == element:
if node.next:
node.element = node.next.element;
node.next = node.next.next;
else:
print(node.element)
node = None;
break;
node = node.next
def get(self, element):
node = self.head
while node:
if node.element == element:
return node
node = node.next
return None
我正在测试 remove 方法,当我尝试删除链表的最后一个元素时它不起作用。在这种情况下,最后一个元素应设置为"None",如下面的代码片段所示(包含在上面的类中(。
def remove(self, element):
node = self.head
while node:
if node.element == element:
if node.next:
node.element = node.next.element;
node.next = node.next.next;
else:
node = None; #this will be fired
break;
node = node.next
但是,最后一个元素没有被删除,我不确定为什么。我假设node
将是指向最后一个元素的指针,并且该值将设置为 none
,但是它不会被删除。以下是失败的测试:
class Test(unittest.TestCase):
def test_remove_node(self):
head = Node('a', Node('b', Node('c')));
ll = LinkedList(head);
ll.remove('c');
self.assertEqual(head.element, 'a');
self.assertEqual(head.next.element, 'b');
print(head.next.next.element);
self.assertIsNone(head.next.next);
if __name__ == "__main__":
unittest.main()
node
是一个临时局部变量,其中包含对感兴趣节点的附加引用。将其设置为 None
不会改变列表的结构,就像在循环的每次迭代中更新列表一样:
node
v
head -> a -> b -> c -> None
与
node
v
head -> a -> b -> c -> None
若要正确更改列表的内容,必须更新上一个节点中的指针。您可以通过在迭代时跟踪父级来执行此操作:
def remove(self, element):
node, parent = self.head, None
while node:
if node.element == element:
if parent:
parent.next = node.next
else:
self.head = node.next
break
node, parent = node.next, node
您永远不需要复制元素,也永远不需要查看过去node.next
。只需绕过包含感兴趣元素的节点就足够了。