这是一道关于链表的题目。
【题目描述】
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回新的头节点 。
题目来源(力扣):移除链表元素
示例1:
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
示例2:
输入:head = [ ], val = 1
输出:[ ]
示例3:
输入:head = [7,7,7,7], val = 7
输出:[ ]
【解题思路】
对本题采用先后指针法,先将 prev 指针初始化为空指针,将 cur 指针指向头节点。
下面是一个循环体:
由此可分为两大情况:cur 指向节点的值是否为 val 。
若不是,将前后指针都后移一个节点;
若是,则又可以分为两种情况:该节点是否为头节点。
若该节点是头节点,就移除头节点,让下一个节点变为新的头节点;若该节点不是头节点,则相当于移除中间节点。
最后考虑循环条件,显然,当 cur 指针为空指针时,循环结束,所以循环条件是 cur 为非空。
【代码实现】
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode* prev = NULL;
struct ListNode* cur = head;
while(cur)
{
if(cur->val == val)
{
//移除头节点
if(cur == head)
{
head = cur->next;
free(cur);
cur = head;
}
else
{
prev->next = cur->next;
free(cur);
cur = prev->next;
}
}
else
{
prev = cur;
cur = cur->next;
}
}
return head;
}
给的一般示例和极端示例都能通过。
比较巧的地方是前后指针的初始化条件:先将 prev 指针初始化为空指针,将 cur 指针指向头节点。
而不是 将 prev 指针指向头节点,将 cur 指针指向头节点的下一个节点。因为这样做很难解决问题。
其它专栏
有关数组的题目