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

Question

给定一个排序链表，删除所有重复编号的节点，只留下原始列表中不同的编号。

示例 1：

Input: 1->2->3->3->4->4->5
Output: 1->2->5

示例 2：

Input: 1->1->1->2->3
Output: 2->3

我已经在大多数情况下尝试并成功地编写了我的代码，我唯一缺少的情况是列表以重复结尾并且整个过程中没有重复。

class Solution:
    def deleteDuplicates(self, head: ListNode) -> ListNode:
        first = head
        if first is None:
            return []
        second = head.next
        if second is None:
            return first
        first.next = second
        if first.val == first.next.val:
            while first.val == first.next.val:
                if first.next.next is None:
                    return []
                first = first.next
            return self.deleteDuplicates(first.next)
        else:
            first.next = self.deleteDuplicates(first.next)
            return first

我得到的 [1,2,3,4,4] 错误是“AttributeError: 'list' object has no attribute 'val'”。

Answer 1

假设你的数据是可散列的，你可以在 $O(n)$ 时间和 $O(n)$ 空间（最坏情况）内完成此操作，只需跟踪你目前看到的内容。

from collections import defaultdict

def mark_duplicates(current_node, seen_so_far):
    next = current_node.next
    seen_so_far[current_node.data] += 1
    if next is None: return seen_so_far
    return mark_duplicates(next, seen_so_far)

def remove_duplicates(current_node, seen):
    next = current_node.next
    prev = current_node.prev
    if seen[current_node.data] > 1:
        if prev is not None: prev.next = next
        if next is not None: next.prev = prev
        # No need to delete current_node, the GC will do it
    if next is not None: remove_duplicates(next, seen)

notes = mark_duplicates(head_of_list, defaultdict(int))
remove_duplicates(head_of_list, notes)

defaultdict(int) 只是一个字典，当您尝试访问不存在的键时，它会返回 0。所以这会计算每个值出现的次数，然后删除出现多次的所有内容。

Answer 2

代码工作正常，我只需将 return [] 更改为简单地“return”，因为 [] 没有被识别为空链表，而只是一个空链表。感谢 Devesh Kumar Singh 认识到这个错误，但他没有将其作为答案发布，所以我代表他发布。