题目
给你一个链表的头节点 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-linked-list-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
不使用虚拟表头的话,当删除的结点处于表头时,需要单独处理,所以需要多一道程序;
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode *temp;
while(head && head->val == val){
temp = head;
head = head->next;
free(temp);
}
struct ListNode *front,*back;
for(front=head;front;front=back){
back = front->next;
while(back && back->val == val){
front->next = back->next;
free(back);
back = front->next;
}
}
return head;
}
使用虚拟表头,程序将简化不少
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode dummyHead;
dummyHead.next = head;
struct ListNode* p = &dummyHead;
struct ListNode* temp;
while(p->next){
if(p->next->val ==val){
temp = p->next;
p->next = p->next->next;
free(temp);
}else{
p = p->next;
}
}
return dummyHead.next;
}
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode* dummyHead = malloc(sizeof(struct ListNode));
dummyHead->next = head;
struct ListNode *p,*q;
for(p=dummyHead;p;p=q){
q=p->next;
while(q && q->val == val){
p->next = q->next;
free(q);
q = p->next;
}
}
head = dummyHead->next;
free(dummyHead);
return head;
}
知识点小结
链表中使用虚拟表头将会简化很多问题