LeetCode 203. Remove Linked List Elements 移除链表元素 C++/Java

Remove all elements from a linked list of integers that have value val.

Example:

Input:  1->2->6->3->4->5->6, val = 6
Output: 1->2->3->4->5

本题的思路很简单，就是单纯的删除链表元素操作，如果头结点的元素就是给定的val值，需要借助虚结点dummy去判断。

方法一：（C++） C++中注意结点的释放

 

ListNode* removeElements(ListNode* head, int val) {
        ListNode* dummy=new ListNode(-1),* pre=dummy;
        pre->next=head;
        while(pre->next){
            if(pre->next->val==val){
                ListNode* t=pre->next;
                pre->next=pre->next->next;
                t->next=NULL;
                delete t;
            }
            else
                pre=pre->next;
        }
        return dummy->next;
    }

 

Java：

public ListNode removeElements(ListNode head, int val) {
        if(head==null||(head.next==null&&head.val==val))
            return null;
        ListNode dummy=new ListNode(-1),cur=dummy;
        cur.next=head;
        while(cur.next!=null){
            if(cur.next.val==val)
                cur.next=cur.next.next;
            else
                cur=cur.next;
        }
        return dummy.next;
    }

方法二：不使用虚结点的话，最后只需要再判断一下头结点的值是否符合要求（C++）

ListNode* removeElements(ListNode* head, int val) {
        if(head==NULL)
            return head;
        ListNode* cur=head;
        while(cur->next){
            if(cur->next->val==val)
                cur->next=cur->next->next;
            else 
                cur=cur->next;
        }
        return (head->val==val)?head->next:head;
    }