删除链表的节点
题目:给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。返回删除后的链表的头节点。
注意:此题对比原题有改动
示例 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 即可。
- 初始化双指针: r =NULL , p = head->next 。
- 定位节点: 当 p 为空 或 p 节点值等于 val 时跳出。
- 保存当前节点索引,即 r = p 。遍历下一节点,即 p = p->next.
- 删除节点: 若 p 指向某节点,则执行 r->next =r->next->next 。(若 p 指向NULL ,代表链表中不包含值为 val 的节点。)
- 返回值: 返回链表头部节点 head 即可。
代码展示
代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* deleteNode(ListNode* head, int val) {
ListNode *p=head,*r=NULL;
if(head==NULL) return head;
if(head->val==val) return head->next;
while(p && p->val!=val)
{
r=p;
p=p->next;
}
r->next=r->next->next;
return head;
}
};
执行用时:4 ms, 在所有 C++ 提交中击败了99.85%的用户
内存消耗:8.2 MB, 在所有 C++ 提交中击败了98.65%的用户