删除变量(指针)值不会删除它指向的值

时间:2019-04-01 04:30:21

标签: python linked-list

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

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

1 个答案:

答案 0 :(得分:0)

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