刪除鏈表中等于給定值val的所有節(jié)點(diǎn)侈沪。
您在真實(shí)的面試中是否遇到過這個(gè)題份帐?
Yes
樣例
給出鏈表 1->2->3->3->4->5->3, 和 val = 3, 你需要返回刪除3之后的鏈表:1->2->4->5。
class Solution {
public:
/**
* @param head a ListNode
* @param val an integer
* @return a ListNode
*/
ListNode *removeElements(ListNode *head, int val) {
// Write your code here
//考慮頭節(jié)點(diǎn)是val
while(head==NULL||head->val==val){
if(head==NULL){
return head;
}
else{
head=head->next;
}
}
//如果頭節(jié)點(diǎn)不是的話
ListNode * pre=head;
ListNode * p=head->next;
while(p!=NULL){
if(p->val==val){
pre->next=p->next;
// ListNode * del=p;
// delete del;
p=p->next;
}
else{
pre=pre->next;
p=p->next;
}
}
return head;
}
};