LeetCode----Remove Duplicates from Sorted List II

Remove Duplicates from Sorted List II

Given a sorted linked list, delete all nodes that have duplicate numbers, leaving only distinct numbers from the original list.

For example,
Given 1->2->3->3->4->4->5, return 1->2->5.
Given 1->1->1->2->3, return 2->3.

分析：

删除链表中所有重复的元素。

使用集合记录所有重复过的元素，然后再次遍历链表，将不在重复集合中的节点加入新的链表。时间复杂度为O（n），空间也为O（n）。

当然，也可以每次遍历找出非重复的元素，然后添入新的链表，这样空间满足O（1），但是时间为O（n^2）.

代码：

class Solution(object):
    def removeElements(self, head, val):
        """
        :type head: ListNode
        :type val: int
        :rtype: ListNode
        """
        p = head
        v = set()  # record all the values
        d = set()  # record only the duplicate values
        while p:
            if p.val not in v:
                v.add(p.val)
            else:
                d.add(p.val)
            p = p.next
        p = head
        r = q = ListNode(-1)
        while p:
            if p.val not in d:
                q.next = p
                q = q.next
            p = p.next
            q.next = None
        return r.next