无法删除链接列表中的第一个节点

Question

我想在python中创建一个链接列表类（我毫无意义，但它是一个学习练习），而且我写入的删除节点的方法不起作用如果我尝试删除链接列表的第一个元素。如果要删除的节点是链接列表中的其他任何地方，则该方法工作正常。有人能给我一些见解，看看我哪里出了问题吗？

这里是我的代码迄今：

class Node: 

    def __init__(self, data=None, next=None): 
     self.data = data 
     self.next = next 

    def __repr__(self): 
     return repr(self.data) 

    def printNodes(self): 
     while self: 
      print self.data 
      self = self.next 

    def removeNode(self, datum): 
     """removes node from linked list""" 
     if self.data == datum: 
      return self.next 
     while self.next: 
      if self.next.data == datum: 
       self.next = self.next.next 
       return self 
      self = self.next 

Answer 1

修改removeNode，以便它始终返回链表的首部，然后将结果返回给您的头节点。像这样：

def removeNode(self, datum): 
    """removes node from linked list and returns head node""" 
    head = self 
    curr_node = self 
    if curr_node.data == datum: 
     head = curr_node.next 
    else: 
     while curr_node.next: 
      if curr_node.next.data == datum: 
       curr_node.next = self.next.next 
       break 
      curr_node = curr_node.next 
    return head 

或者，如果你想避免不必的removeNode结果分配回头：

def removeNode(self, datum): 
    """removes node from linked list""" 
    curr_node = self 
    if curr_node.data == datum: 
     # steals the the data from the second node 
     curr_node.data = curr_node.next.data 
     curr_node.next = curr_node.next.next 
    else: 
     while curr_node.next: 
      if curr_node.next.data == datum: 
       curr_node.next = curr_node.next.next 
       break 
      curr_node = curr_node.next 

注：我分配self给curr_node，因为它觉得不对劲修改self。

Answer 2

在你的代码所示，removeNode方法返回的链接列表的新的第一个节点。您应该使用head = head.removeNode(1)而不是只调用该方法。