给你一个链表的头节点 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 输出:[]
提示:
- 列表中的节点数目在范围
[0, 104]
内 1 <= Node.val <= 50
0 <= val <= 50
思路:
题目是把等于val的节点删除,返回一个新节点,那么我们不如换个思路。
把等于val值的节点链接在一起,再返回头节点,不就可以了吗。
对于等于val的节点我们怎么链接呢,这时候需要两个指针来维护新链表的头和尾,下面给出示图及代码。
typedef struct ListNode ListNode;
struct ListNode* removeElements(struct ListNode* head, int val)
{
//创建两个新指针
ListNode *newhead, *newtail;
newhead = newtail = NULL;
//pcur遍历原链表
ListNode* pcur = head;
while (pcur != NULL)
{
if (pcur->val != val)
{
if (newhead == NULL)
{
//头指针为空
newhead = newtail = pcur;
}
else
{
//头指针不为空
newtail->next = pcur;
newtail = newtail->next;
}
}
pcur = pcur->next;
}
//防止最后pcur->next为NULL,出现*NULL或NULL->next操作
if (newtail)
newtail->next = NULL;
return newhead;
}