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ノードに向ける.
/**
 * 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;
    }
};