LintCode【入门】删除链表中的元素

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

样例

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

/** * 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 head;        }        ListNode *p, *q;        p = head;        q = p->next;        while (q!=NULL) {            if (q->val == val) {                q = q->next;                p->next = q;            } else {                p = p->next;                q = q->next;            }        }        return head;    }};