删除排序链表中的重复元素 II

删除排序链表中的重复元素 II

题目描述

Category	Difficulty	Likes	Dislikes
algorithms	Medium (53.63%)	1314	-

• Tags

  linked-list

• Companies

  Unknown

给定一个已排序的链表的头 head ， 删除原始链表中所有重复数字的节点，只留下不同的数字 。返回 已排序的链表 。

示例 1：

输入：head = [1,2,3,3,4,4,5]
输出：[1,2,5]

示例 2：

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

提示：

• 链表中节点数目在范围 [0, 300] 内
• 100 <= Node.val <= 100
• 题目数据保证链表已经按升序 排列

---

求解

方法一：一次遍历

思路与算法

由于给定的链表是排好序的，因此重复的元素在链表中出现的位置是连续的，因此我们只需要对链表进行一次遍历，就可以删除重复的元素。由于链表的头节点可能会被删除，因此我们需要额外使用一个哑节点（dummy node）指向链表的头节点。

具体地，我们从指针 pre 指向链表的哑节点，curr指向pre的下一个节点，随后开始对链表进行遍历。如果当前 cur 与 cur.next 对应的元素相同，那么我们就需要将 cur 以及所有后面拥有相同元素值的链表节点全部删除。具体的，我们将curr和curr.next比较，如果相同则后移curr，直到 curr.next 为空节点或者其元素值不等于curr。此时，我们将pre.next指向curr.next即可。

当遍历完整个链表之后，我们删除哑节点dummy并返回链表的的哑节点的下一个节点 dummy.next 即可。

细节

需要注意 cur 以及 cur.next可能为空节点，如果不加以判断，可能会产生运行错误。

代码

class Solution
{
public:
    ListNode *deleteDuplicates(ListNode *head)
    {
        if (!head || !head->next)
            return head;

        ListNode *dummy = new ListNode(-1, head);
        ListNode *pre = dummy, *curr = head;
        while (curr && curr->next)
        {
            if (curr->val == curr->next->val)
            {
                while (curr->next && curr->val == curr->next->val)
                {
                    curr = curr->next;
                }

                pre->next = curr->next;
                curr = curr->next;
            }
            else
            {
                pre = curr;
                curr = curr->next;
            }
        }

        ListNode *ans = dummy->next;
        delete dummy;
        return ans;
    }
};

复杂度分析

• 时间复杂度：O(n)，其中 n 是链表的长度。
• 空间复杂度：O(1)。