巧用「递归 + 双指针」，斩获 LeetCode「删除排序链表中的重复元素 II」难题

IOS

2024-01-29 19:47:50

在 LeetCode 中，「删除排序链表中的重复元素 II」问题要求我们删除链表中重复出现的元素，并保留其第一次出现的节点。相较于「删除排序链表中的重复元素」问题，本题的难点在于需要删除所有重复出现的元素 。

为了应对这个挑战，本文将介绍一种巧妙的「递归 + 双指针」方法。这种方法通过递归和双指针技术，以优雅的方式解决问题，展现其高效性。

算法详解

「递归 + 双指针」方法

该方法包含以下关键步骤：

创建两个指针：
- current 指针指向当前节点。
- prev 指针指向 current 的前一个节点。
递归处理余下链表：
- 使用递归调用 removeDuplicates() 函数处理当前节点之后的链表。
比较当前元素和下一个元素：
- 如果 current.val 等于 current.next.val，则存在重复。
删除重复元素：
- 如果存在重复，则将 prev.next 指向 current.next.next，从而跳过重复元素。
移动指针：
- 如果不存在重复，则将 prev 和 current 指针向后移动一位。
递归返回：
- 递归返回 prev.next，表示删除重复元素后的链表。

代码实现

def removeDuplicates(head):
    if not head or not head.next:
        return head

    prev = None
    current = head

    while current:
        if current.val == prev.val:
            prev.next = current.next
        else:
            prev = current
        current = current.next

    return head