Question

我正在编码一个链表的实现，链表类及其节点看起来像这样：

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方法，当我尝试删除链表的最后一个元素时，该方法不起作用。在这种情况下，最后一个元素应设置为“无”，如下面的代码片段所示（该代码包含在上面的类中）。

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()

Answer 1

node是一个临时局部变量，其中包含对所关注节点的附加引用。将其设置为None不会改变列表的结构，就像在每次循环迭代时更新列表一样：

                node
                  v
head -> a -> b -> c -> None

vs

                       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。仅仅绕过包含感兴趣元素的节点就足够了。

删除变量（指针）值不会删除它指向的值

1 个答案: