def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
时间: 2024-10-24 13:13:55 浏览: 18
`removeElements` 函数用于从给定的链表(`head`)中删除所有值等于 `val` 的元素,返回新链表的头节点。这个函数采用迭代方式实现:
1. 创建一个虚拟头节点 `dummy`,并将其 `next` 指向原始链表的头 `head`。
2. 使用一个指针 `cur` 初始化为 `dummy`,然后开始遍历链表。
3. 在每次循环中,检查当前节点 `cur.next` 的值是否等于 `val`。如果相等,则跳过该节点,通过设置 `cur.next` 为 `cur.next.next` 来移除它。
4. 如果不等于 `val`,则简单地移动 `cur` 到下一个节点。
5. 循环结束后,`dummy.next` 就指向了新的链表头,所以返回 `dummy.next`。
代码示例如下[^1]:
```python
def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
dummy = ListNode() # 创建虚拟头节点
dummy.next = head # 将 dummy 的 next 指向原头
cur = dummy # 定义当前节点
while cur and cur.next: # 当还有节点时继续
if cur.next.val == val: # 找到要删除的节点
cur.next = cur.next.next # 跳过它
else:
cur = cur.next # 否则,移动到下一个节点
return dummy.next # 返回新链表的头节点
```
阅读全文