题目描述
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点,返回删除后的链表的头节点。
示例 1:
输入: head = [4,5,1,9], val = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.
解题思路
1. 双指针法
本题删除值为 val 的节点分需为两步:定位节点、修改引用。
- 定位节点: 遍历链表,直到 head.val == val 时跳出,即可定位目标节点。
- 修改引用: 设节点 cur 的前驱节点为 pre,后继节点为 cur.next ;则执行 pre.next = cur.next ,即可实现删除 cur 节点。
算法流程
- 特例处理: 当应删除头节点 head 时,直接返回 head.next 即可。
- 初始化: pre = head , cur = head.next 。
- 定位节点: 当 cur 为空 或 cur 节点值等于 val 时跳出。
- 保存当前节点索引,即 pre = cur 。
- 遍历下一节点,即 cur = cur.next 。
- 删除节点: 若 cur 指向某节点,则执行 pre.next = cur.next ;若 cur 指向 null,代表链表中不包含值为 val 的节点。
- 返回值: 返回链表头部节点 head 即可。
复杂度分析
- 时间复杂度 O(N) : NN为链表长度,删除操作平均需循环 N/2 次,最差 N 次。
- 空间复杂度 O(1): cur, pre 占用常数大小额外空间。
代码
class Solution {
public ListNode deleteNode(ListNode head, int val) {
if(head.val == val) return head.next;
ListNode pre = head, cur = head.next;
while(cur != null && cur.val != val) {
pre = cur;
cur = cur.next;
}
if(cur != null) pre.next = cur.next;
return head;
}
}
2.递归法
递归思路
- 第一步,定义递归函数:
public ListNode deleteNode(ListNode head, int val)
- 确定终止条件:上述函数表示的是删除链表中值为val的结点,当链表为空时,我们无法做删除操作,所以递归的终止条件是head == null
if (head == null) {
return null;
}
- 如果head结点不等于空,并且head结点的值等于val,我们直接返回head结点的下一个结点
if (head.val == val) {
return head.next;
}
- 否则也就是说头结点是删不掉的,我们就递归调用,从头结点的下一个开始继续上面的操作,直到删除为止。
else {
head.next = deleteNode(head.next, val);
}
return head;
代码
class Solution {
public ListNode deleteNode(ListNode head, int val) {
if (head == null) {
return null;
}
if (head.val == val) {
return head.next;
} else {
head.next = deleteNode(head.next, val);
}
return head;
}
}