删除链表中的元素

描述

删除链表中等于给定值val的所有节点。

样例

给出链表 1->2->3->3->4->5->3, 和 val = 3, 你需要返回删除3之后的链表：1->2->4->5。

思考

1. 是否为空
2. 对头结点进行判断
3. 对后续结点进行判断

代码

//  By Lentitude


/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    /**
     * @param head a ListNode
     * @param val an integer
     * @return a ListNode
     */
    ListNode *removeElements(ListNode *head, int val) {
        // Write your code here
        // 判断链表是否为空
        if (head == NULL) return head;

        // 对头结点进行处理
        while (head->val == val){
            head = head ->next;
            if (head == NULL){
                return NULL;
            }
        }

        // 对后面的节点进行处理
        ListNode *pre = head;
        while (pre->next != NULL){
            if (pre->next->val == val){
                pre->next = pre->next->next;
            }else{
                pre = pre->next;
            }
        }

        return head;
    }
};