删除链表中等于给定值 val 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
方法1:如果要删掉一个节点的话,那么就令其前一个节点指向其后面的节点。设置pre,p两个节点,分别代表前一个节点、当前节点。为了处理被删除节点是头节点的情况,我们在头节点前面设置一个虚拟节点dummy。
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode* dummy=new ListNode(0), *p=head, *pre=dummy;
dummy->next=head;
while(p){
if(p->val==val){
pre->next=p->next;
p=p->next;
}
else{
pre=p;
p=p->next;
}
}
return dummy->next;
}
};
方法2:用递归
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
if(!head) return nullptr;
head->next= removeElements(head->next, val);
return head->val==val ? head->next :head;
}
};