203-Remove Linked List Elements

難度：easy

類别：linked list

1.題目描述

Remove all elements from a linked list of integers that have value val.

Example

Given: 1 –> 2 –> 6 –> 3 –> 4 –> 5 –> 6, val = 6

Return: 1 –> 2 –> 3 –> 4 –> 5

2.算法分析

周遊連結清單，比較head->next的值與val是否相等，如果相等的話，将其删除，并且head->next = head->next->next.

因為while循環中是head->next，是以最後要對第一節點進行判斷。

整體思路很簡單，實作也不難。

3.代碼實作

ListNode* removeElements(ListNode* head, int val) {
    if (head == NULL) return head;
    if (head->next == NULL && head->val == val) return NULL;
    if (head->next == NULL && head->val != val) return head;
    ListNode* result = head;
    ListNode* temp = NULL;
    while (head->next != NULL) {
        if (head->next->val == val) {
            temp = head->next;
            head->next = temp->next;
            delete temp;
        }
        else {
            head = head->next;
        }
    }
    // 最後的時候再判斷第一節點，看是不是需要删除
    if (result->val == val) {
        result = result->next;
    }
    return result;
}