面试金典10（Python）—— 移除重复节点（简单）

移除重复节点

概述：编写代码，移除未排序链表中的重复节点。保留最开始出现的节点。

输入：[1, 2, 3, 3, 2, 1]
输出：[1, 2, 3]

输入：[1, 1, 1, 1, 2]
输出：[1, 2]

方法一：哈希表+循环

思路：首先注意到链表是未排序的，那么可以建立一个哈希表 head_dict 进行存储判断。若不存在，添加到哈希表中，并指向下一个节点，一直循环结束即可。

# 哈希表+循环
class Solution:
    def removeDuplicateNodes(self, head: ListNode) -> ListNode:
        if not head:
            return head
        head_dict = {head.val}
        pos = head
        while pos.next:
            temp = pos.next
            if temp.val not in head_dict:
                head_dict.add(temp.val)
                pos = pos.next
            else:
                pos.next = pos.next.next
        return head

方法二：集合+循环

思路：思路基本上是一致的，不同在于该算法用 set() 处理未排序问题。最后定义双指针，或叫快慢指针均可，来循环判断即可。

# 集合+循环
class Solution:
    def removeDuplicateNodes(self, head: ListNode) -> ListNode:
        pre, cur = None, head
        visited = set()
        while cur:
            if cur.val in visited:
                pre.next = cur.next
            else:
                visited.add(cur.val)
                pre = cur
            cur = cur.next
        return head

方法三：二重循环（会超时）

思路：进阶要求是，不使用临时缓存，那么从头节点开始二重循环即可。简单点说就是，时间换空间。

# 二重循环（会超时）
class Solution:
    def removeDuplicateNodes(self, head: ListNode) -> ListNode:
        cur = head
        while cur:
            pos = cur
            while pos.next:
                if pos.next.val == cur.val:
                    pos.next = pos.next.next
                else:
                    pos = pos.next
            cur = cur.next
        return head

总结

我以为标个进阶是啥精妙的解法，结果说了个暴力循环！