LeetCode 203. Remove Linked List Elements(C++版)
1233 ワード
Remove all elements from a linked list of integers that have value val.
ExampleGiven:1-->2-->6-->3-->4-->5-->6、val=6 Return:1-->2-->3-->4-->5構想分析:
チェーンテーブルを遍歴し、現在遍歴しているチェーンテーブルのノードがpであることを記し、p->val==valであれば、そのノードを削除し、ノードを削除するのは2つの状況に分けられる.
1)削除したのはヘッドノードで、直接ヘッドポインタを後ろに移動します.
2)他のノードを削除する場合は,pノードの前のノードpreのnextをpのnextに変更する必要がある.次にpをこのときpreのnextノードに向ける.
ExampleGiven:1-->2-->6-->3-->4-->5-->6、val=6 Return:1-->2-->3-->4-->5構想分析:
チェーンテーブルを遍歴し、現在遍歴しているチェーンテーブルのノードがpであることを記し、p->val==valであれば、そのノードを削除し、ノードを削除するのは2つの状況に分けられる.
1)削除したのはヘッドノードで、直接ヘッドポインタを後ろに移動します.
2)他のノードを削除する場合は,pノードの前のノードpreのnextをpのnextに変更する必要がある.次にpをこのときpreのnextノードに向ける.
/**
* 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) {
ListNode *p = head;
ListNode *pre = head;
while(p){//p
if(head -> val == val){
head = head -> next;
free(p);
p = head;
pre = head;
continue;
}
if(val == p -> val){
pre -> next = p -> next;
free(p);
p = pre -> next;
continue;
}
pre = p;
p = p -> next;
}
return head;
}
};