題目:刪除鏈表中等于給定值 val 的所有節(jié)點。
示例:
輸入: 1->2->6->3->4->5->6, val = 6
輸出: 1->2->3->4->5
解題思路:
1.不帶頭節(jié)點的鏈表,要首先處理頭節(jié)點;
2.要判斷頭節(jié)點是否為空,否則會報錯,顯示:訪問空結(jié)點。
代碼:
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode *p, *pre; //不帶頭結(jié)點
while(head) //處理頭節(jié)點
{
if(head->val == val)
{
head = head->next;
}
else
break;
}
if(head != NULL) //防止訪問空指針
{
pre = head;
p =pre->next;
while(p)
{
if(p->val == val)
{
pre->next = p->next;
p = pre->next;
}
else
{
pre = p;
p = pre->next;
}
}
}
return head;
}