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
- 当 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, ]
广告
数据结构
网络
关系数据库管理系统(RDBMS)
操作系统
Java
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP