题目大意:给出一个链表,和一个要移除元素,返回移除了所有元素的链表。
算法思想:
1.首先,判断表头的元素与所给元素是否相等,采用循环的方式移除所有与表头元素。
2.采用前驱指针的方法,通过遍历链表,移除链表中符合要求的元素。
代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
if(head==NULL) return NULL;
while(head->val==val){
ListNode* temp=head;
if(head->next==NULL){
delete head;
return NULL;
}
head=head->next;
delete temp;
}
ListNode *pre=head,*cp=head->next;
while(cp!=NULL){
if(cp->val==val){
pre->next=cp->next;
delete cp;
cp=pre->next;
}
else{
pre=cp;
cp=cp->next;
}
}
return head;
}
};