面试题 02.01. 移除重复节点

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

示例1:

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

1、题目分析

题目比较清晰，就是删除重复节点。可以利用一个set集合对重复节点去重。

2、解题分析

• 判断链表是否为空，不为空进行下一步
• 声明一个set集合，用来存储出现过的节点
• 如果节点的值在集合中就做删除节点pos.next = pos.next.next
• 如果节点不在集合中，添加节点的值到集合中并继续遍历pos = pos.next

这个代码中pos是指向head的当前节点，然后所有操作的都是对原链表产生影响和变化。

3、代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeDuplicateNodes(self, head: ListNode) -> ListNode:
        if not head:
            return head
        s = {head.val}
        #pos指向head的第一个节点
        pos = head
        while pos.next:
            cur = pos.next
            if cur.val not in s:
                s.add(cur.val)
                pos = pos.next
            else:
                pos.next = pos.next.next
        return head


                

总结：看到重复两个字必定想到用hash或者set。