例题描述
删除单链表中所有等于某给给定值的所有结点。
示例 :
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
结构体定义
//创建结点的结构体
struct ListNode {
int val;
struct ListNode *next;
};
解题思路
单链表正常的概念表现形式:
此时抽象成为一下图片,将结点的值罗列在一起模拟链表,便于演示:
- 利用断言排除掉链表中无结点,即空链表的情况,此时无法删除任何一个结点,所以直接返回
NULL
。 - 创建一个
cur
当前指针来保存当前结点的指针,初始化为头指针的next
,即为指向下一结点的指针。 - 创建一个
prev
前驱指针来保存前一个结点的指针,旨在保存当前指针的前一个结点的指针,方便链接截断结点的链表,初始化为头(head
)指针。
- 建立循环遍历链表结点,判断当前指针
cur
是否为空。如果为空,说明此时已经到达链表尾部,则跳出循环。如果不为空,进行结点值比较。 - 用
cur
指针所指向结点内的值,与所给定的某一数值进行比较,如果不相同,那么就继续检查下一个结点。(如图所示,此时2
与所给定的6
并不相等) - 不相等所执行的逻辑:
prev
前驱指针的指向变更为cur
当前指针
cur
当前指针指向下一个结点:
此时,完成了一轮数据的更迭,继续下一次循环。
- 再次进行判断
cur
当前指针是否为空,非空说明没有到达表尾,继续比较,此时我们发现cur
所指向的结点值与给定值相等,执行删除逻辑。
删除逻辑的思路为:让prev前驱指针的下一结点指向变更为cur当前结点的下一结点指向,即跳过了cur
当前指针所指向的结点,变相地完成了“删除”逻辑。
- 完成删除逻辑之后还要进行下一次循环,并及时收回没有使用的内存,我们选择
free(cur)
释放cur
当前指针指向的内存。那么此时prev
指针记录下了上一个结点的位置,将prev
指针指向cur
指针的下一个结点可以链接删除结点的前后半段链表。
- 按照逻辑继续执行,当
cur
当前指针到达了链表尾部,即指向了NULL
,此时循环不再满足而跳出,完成删除逻辑。此时链表已经删除了等于给定某值的所有结点。
- 【特殊处理】: 如果链表中只有一个结点,这个结点的值与给定值相等,即要删除的结点,那么此时就上面的逻辑就无法满足此情景,需要在下面特殊处理一下单个结点的情况:
【创建一个"自己的”头指针指向链表的头指针,判断头指针指向的这一个结点是否是要删除的结点,如果不是,直接返回我们自己得头指针即可。如果是,使我们的头指针指向链表头指针的下一个结点(此时为NULL
),跳过这个结点即可】
使用自建的头指针可以将两种情况规整在同一代码中
代码实现
struct ListNode* removeElements(struct ListNode* head, int val) {
if (head == NULL) {
return NULL;
}
struct ListNode *prev = head;
struct ListNode *cur = head->next;
while (cur != NULL) {
if (cur->val != val) {
prev = cur;
}
else {
prev->next = cur->next;
free(cur);
}
cur = prev->next;
}
struct ListNode *NewHead = head;
if (head->val == val) {
NewHead = head->next;
free(head);
}
return NewHead;
}
这里提供下篇博客链接,此博客介绍了删除顺序表中等于给定值 val 的所有结点的思路,有兴趣的读者可以点击蓝色字体跳转至下篇博客。