文章目录
- 题目描述
- 思路一:
- 思路二:
题目描述
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
示例:
思路一:
要移除链表中值为val的结点,我们肯定是要将链表遍历一遍,关键是我们在遍历的过程中应该如何操作。我们考虑问题的时候,可以先考虑比较常见的情况,再考虑特殊情况。
一、考虑常见情况
要移除某一结点,也就是让该结点的前一个结点指向待移除结点的后一个结点,然后将待移除结点释放即可。我们可以定义3个指针变量:prev,cur,next 。
prev:记录待排查结点的前一个结点位置(previous)。
cur:记录当前正在排查的结点位置(current)。
next:记录待排查结点的后一个结点(next)。
当cur指针指向的结点并非待移除的结点时,3个结点依次向后移动。
当cur指针指向待移除的结点时,我们首先让prev指针指向的结点指向next,然后将cur指针指向的结点释放掉,并将next指针赋值给cur指针,next指针再后移。
如此进行下去,直到链表遍历完毕,那么值为val的结点也就删除了。
二、考虑特殊情况
常见情况的分析往往只能解决问题的一般情况,并不能解决问题的极端情况。要真正解决问题,我们需要考虑到问题的极端情况。例如,当待移除的结点是第一个结点或是最后一个结点的情况,当链表为空的情况。
1.当链表的第一个结点为待移除的结点时
这时我们需要先将头指针指向next,然后释放cur指向的结点,并将next指针赋值给cur指针,next指针再后移。
2.当链表的最后一个结点为待移除的结点时
当排查到最后一个结点时,cur指向最后一个结点,next指针指向该结点指向的位置,即NULL。
我们用上面常规情况的方法对其进行分析,发现常规情况的思路适用于这种特殊情况。并且发现遍历的终止条件,就是当cur为NULL的时候遍历停止。
3.当传入链表为空时
我们可以发现,若传入的链表为空链表(NULL),cur指针的值一开始就为空,而我们遍历的终止条件就是当cur为NULL时停止遍历,所以当传入链表为空时,直接执行到函数末尾,即返回头指针(NULL)。
代码示例:
struct ListNode {
int val;
struct ListNode *next;
};
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode* prev = NULL;//记录待排查结点的前一个结点位置
struct ListNode* cur = head;//记录当前正在排查的结点位置
while (cur != NULL)//当cur为空时,循环停止
{
if (cur->val == val)//当前排查的结点是待移除的结点
{
struct ListNode* next = cur->next;//记录待排查结点的后一个结点位置
if (cur == head)//待移除的结点是链表的第一个结点
{
head = next;//头指针指向next
free(cur);//释放第一个结点
cur = next;//将next指针赋值给cur指针
}
else//待移除的结点不是链表的第一个结点
{
prev->next = next;//prev指针指向的结点指向next
free(cur);//将cur指针指向的结点释放掉
cur = next;//将next指针赋值给cur指针
}
}
else//当前排查的结点不是待移除的结点
{
prev = cur;//指针后移
cur = cur->next;//指针后移
}
}
return head;//返回新的头指针
}
思路二:
我们可能觉得思路一的代码比较复杂,当我们要移除某一个结点时,还需要判断该结点是否为第一个结点,那么有没有什么办法可以不用进行这一步操作呢?
回答是肯定的。办法就是在传入的链表前面强行加上一个头结点,并让链表原来的头指针指向该头结点,这样我们就不用判断待移除的结点是否为第一个结点了(因为现在第一个结点是头结点)。
在加了头结点后,我们就只需要根据常见情况的逻辑进行代码的编写即可。但是有一点不能忘记,就是在遍历完链表后要将头结点指向的位置(即第一个结点的位置)赋值给头指针,并将头结点释放掉,最后才能返回头指针。
代码示例:
struct ListNode {
int val;
struct ListNode *next;
};
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode* guard = (struct ListNode*)malloc(sizeof(struct ListNode));//申请一个头结点,返回其地址
guard->next = head;//让头结点指向链表的第一个结点
struct ListNode* cur = guard->next;//cur指针指向原链表第一个结点
struct ListNode* prev = guard;//prev指针指向头结点
while (cur != NULL)//当cur为空时,循环停止
{
if (cur->val == val)//当前排查的结点是待移除的结点
{
struct ListNode* next = cur->next;//记录待排查结点的后一个结点位置
prev->next = next;//prev指针指向的结点指向next
free(cur);//将cur指针指向的结点释放掉
cur = next;//将next指针赋值给cur指针
}
else//当前排查的结点不是待移除的结点
{
prev = cur;//指针后移
cur = cur->next;//指针后移
}
}
head = guard->next;//将头结点指向的位置赋值给头指针,使头指针指向链表第一个结点
free(guard);//释放头结点
guard = NULL;//及时置空
return head;//返回新的头指针
}