[LintCode]452.删除链表中的元素

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

 

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

思路：1.判断head是否为空

           2.要删除的节点在头部

           3.要删除的节点在中间、尾部，遍历链表删除

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

/**
 * 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) {
        if(head==NULL)
            return head;
        ListNode *p=head;
        ListNode *newHead=new ListNode(0);
        ListNode *node=newHead;
        while(p!=NULL){
            if(p->val!=val)
            {
                node->next=p;
                node=node->next;
            }
            p=p->next;
        }
        node->next=NULL;
        return newHead->next;
    }
};