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;
    }
};