Python程序：删除链表中所有值为特定值的节点

假设我们有一个单链表和一个目标值，我们需要删除所有值为目标值的节点后返回该链表。

例如，如果输入是 [5,8,2,6,5,2,9,6,2,4]，那么输出将是 [5, 8, 6, 5, 9, 6, 4]

为了解决这个问题，我们将遵循以下步骤：

  • head := node
  • 当 node 和 node.next 不为空时，执行以下操作：
    • 当 node 的 next 节点的值等于目标值时，执行以下操作：
      • node.next := node.next.next
    • node := node.next
  • 如果 head 的值等于目标值，则：
    • 返回 head.next
  • 否则：
    • 返回 head

让我们来看下面的实现，以便更好地理解：

示例

class ListNode:
   def __init__(self, data, next = None):
      self.val = data
      self.next = next
def make_list(elements):
   head = ListNode(elements[0])
   for element in elements[1:]:
      ptr = head
   while ptr.next:
      ptr = ptr.next
      ptr.next = ListNode(element)
   return head
def print_list(head):
   ptr = head
   print('[', end = "")
   while ptr:
      print(ptr.val, end = ", ")
      ptr = ptr.next
      print(']')
class Solution:
   def solve(self, node, target):
      head = node
      while (node and node.next):
         while node.next.val == target:
            node.next = node.next.next
            node = node.next
         if head.val == target:
            return head.next
         else:
      return head
ob = Solution()
head = make_list([5,8,2,6,5,2,9,6,2,4])
ob.solve(head, 2)
print_list(head)

输入

[5,8,2,6,5,2,9,6,2,4]

输出

[5, 8, 6, 5, 9, 6, 4, ]

更新于：2020年10月6日

350 次浏览

开启你的职业生涯

完成课程获得认证

开始学习
广告