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

给定一个已排序的链表的头 head , 删除原始链表中所有重复数字的节点,只留下不同的数字 。返回 已排序的链表 。
输入:head = [1,2,3,3,4,4,5]
输出:[1,2,5]
输入:head = [1,1,1,2,3]
输出:[2,3]

思路与算法
由于给定的链表是排好序的,因此重复的元素在链表中出现的位置是连续的,因此我们只需要对链表进行一次遍历,就可以删除重复的元素。由于链表的头节点可能会被删除,因此我们需要额外使用一个哑节点(dummy node)指向链表的头节点。
具体地,我们用 cur 指向链表的哑节点,随后开始对链表进行遍历。
如果当前 cur.next 与 cur.next.next 对应的元素相同,那么我们就需要将 cur.next 以及所有后面拥有相同元素值的链表节点全部删除。我们记下这个元素值 x,随后不断判断是否cur.next == x, 将等于x的节点从链表中移除,直到为空节点或者其元素值不等于x为止。注意移除动作是cur.next = cur.next.next, 因此cur的位置是不动的。这时需要继续判断cur.next与cur.next.next对应元素是否相同。

如果当前 cur.next与 cur.next.next对应的元素不相同,那么说明链表中只有一个元素值为 cur.next的节点,那么我们就可以将cur 指向 cur.next。

当遍历完整个链表之后,我们返回链表的的哑节点的下一个节点 即可。

时间复杂度O(n)
空间复杂度O(1)

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def deleteDuplicates(self, head: ListNode) -> ListNode:
        start = ListNode()
        start.next = head
        
        cur = start
        while cur.next and cur.next.next:
            if cur.next.val == cur.next.next.val:
                candidate = cur.next.val
                while cur.next:
                    if cur.next.val == candidate:
                        cur.next = cur.next.next
                    else:
                        break
            else:
                cur = cur.next
        return start.next
        

你可能感兴趣的:(开发,链表,leetcode,数据结构)