203. 移除链表元素-c语言做题满分代码
给你一个链表的头节点 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
输出:[]
对于这一题,其实,很多人觉得可能挺简单的,这题的难点并不是在于算法思想的实现,而在于容错的解决,如果处理不好删除节点后的链表链接问题,就很难拿满分,需要做很多容错处理,解题代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* delete_val(struct ListNode* head,int val){
if(head==NULL){
return NULL;
}
while(head->val==val){
head=head->next;
if(head==NULL){
break;
}
}
if(head==NULL){
return NULL;
}
if(head->next==NULL){
return head;
}
struct ListNode *pre=head,*p=head->next;
while(p){
if(p->val==val){
pre->next=p->next;
}
else{
pre=p;
}
p=p->next;
}
return head;
}
struct ListNode* removeElements(struct ListNode* head, int val){
return delete_val(head,val);
}