LeetCode Remove Linked List Elements 删除链表元素

 

 

题意：移除链表中元素值为val的全部元素。

思路：算法复杂度肯定是O(n)，那么就在追求更少代码和更少额外操作。我做不出来。

 

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        while(head&&head->val==val)    head=head->next;
        if(!head)    return 0;
        ListNode *tmp=head;
        while(head&&head->next)
        {
            if(head->next->val==val)
                head->next=head->next->next;
            else
                head=head->next;
        }
        return tmp;
    }
};