给定一个已排序的链表的头 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