题目
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。
返回删除后的链表的头节点。
注意:此题对比原题有改动
示例 1:
输入: head = [4,5,1,9], val = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.
示例 2:
输入: head = [4,5,1,9], val = 1
输出: [4,5,9]
解释: 给定你链表中值为 1 的第三个节点,那么在调用了你的函数之后,该链表应变为 4 -> 5 -> 9.
说明:
- 题目保证链表中节点的值互不相同
- 若使用 C 或 C++ 语言,你不需要
free
或delete
被删除的节点
解题思路
算法步骤:
特例处理: 当应删除头节点 head 时,直接返回 head.next 即可。
初始化: cur = head。
定位节点: 当 cur.next 非空,并且 cur.next 的节点值不等于 val 时进入循环,跳到下一节点:cur = cur.next 。
当 cur.next 非空(即不是尾节点末尾时),表示找到了要删除的节点,那么就更改引用关系:执行 cur.next = cur.next.next。(若跳出循环时 cur.next = null,代表链表中不包含值为 val 的节点。)
返回值: 返回链表头部节点 head 即可。
复杂度分析:
时间复杂度:O(N)。
空间复杂度:O(1)。
注:也可以在链表前面加一个伪头指针,这样就不用再判断原来的第一个头节点是否为要删除的节点了。
代码
解法一:不添加伪头指针
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode deleteNode(ListNode head, int val) {
if(head.val == val){
return head.next;
}
ListNode cur = head;
while(cur.next != null && cur.next.val != val){
cur = cur.next;
}
if(cur.next != null){
cur.next = cur.next.next;
}
return head;
}
}
解法二:添加伪头指针
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode deleteNode(ListNode head, int val) {
ListNode dummy = new ListNode(0);
dummy.next = head;
head = dummy;
while(head.next != null && head.next.val != val){
head = head.next;
}
if(head.next != null){
head.next = head.next.next;
}
return dummy.next;
}
}