18. 删除链表的节点
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点,返回删除后的链表的头节点。
解题思路:
- 删除值为 val 的节点可分为两步:定位节点、删除节点。
- 定位值为 val 的节点: 遍历链表 head ,直到 head.val == val 时跳出,即可定位目标节点。
- 删除链表中某节点: 设 cur 的上个节点为 pre ,下个节点为 cur.next ;则执行 pre.next = cur.next ,即可实现将 cur 从列表删除。
算法流程:
- 特例处理: 当应删除头节点 head 时,直接返回 head.next 即可。
- 初始化: pre = head , cur = head.next 。
- 循环搜索: 当 cur 为空 (即已走过链表尾部节点) 或 cur 节点值等于 val (即找到目标节点) 时跳出。
1.执行 pre = cur : 保存当前节点索引,以便删除节点时使用。
2.执行 cur = cur.next : 遍历下一节点。 - 删除节点: 执行 pre.next = cur.next ,即删除节点 head 。
- 返回值: 返回链表头部节点 head 即可。
复杂度分析:
- 时间复杂度 O(N) : N为链表长度,删除操作平均需循环 N/2 次,最差 N次。
- 空间复杂度 O(1) : cur, pre 占用常数大小额外空间。
代码实现:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def deleteNode(self, head: ListNode, val: int) -> ListNode:
if head.val == val:
return head.next
pre = head
cur = head.next
while cur and cur.val != val:
pre, cur = cur, cur.next
pre.next = cur.next
return head