【leetcode-python】移除重复节点

最新推荐文章于 2022-11-01 18:21:31 发布

原创最新推荐文章于 2022-11-01 18:21:31 发布 · 229 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #python

算法同时被 2 个专栏收录

74 篇文章

订阅专栏

leetcode

65 篇文章

订阅专栏

本文介绍一种算法，用于移除未排序链表中的重复节点，同时保留首次出现的节点。通过遍历链表并使用集合记录已访问节点值的方法，确保每个节点值仅保留一次出现。

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

示例1:

 输入：[1, 2, 3, 3, 2, 1]
 输出：[1, 2, 3]
示例2:

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

链表长度在[0, 20000]范围内。
链表元素在[0, 20000]范围内。

参考解法：
class Solution:
    def removeDuplicateNodes(self, head: ListNode) -> ListNode:
        if not head:
            return head
        node_set = {head.val}
        prev, cur = head, head.next
        while cur:
            if cur.val in node_set:
                prev.next = cur.next
                del cur
                cur = prev.next
            else:
                node_set.add(cur.val)
                prev, cur = cur, cur.next
        return head