LintCode 删除链表中的元素

题目描述：

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

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

思路分析：

水题一枚。

ac代码：

/**
 * 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
        int i=0;
        ListNode *a;
        a=head;
        if(head==NULL)
            return NULL;
        while(head->next!=NULL)
        {
            if(head->next->val==val)
            {
                if(head->next->next==NULL)
                {
                    head->next=NULL;
                    break;
                }
                else
                {
                    head->next=head->next->next;
                }
            }
            else
                head=head->next;
        }
        if(a->val==val)
            return a->next;
        return a;
    }
};