题目
给定一个已排序的链表的头head,删除所有重复的元素,使每个元素只出现一次。返回已排序的链表
示例1:
输入:head = [1,1,2] 输出:[1,2]
示例2:
输入:head = [1,1,2,3,3] 输出:[1,2,3]
提示:
- 链表中节点数目在范围
[0, 300]
内 -100 <= Node.val <= 100
- 题目数据保证链表已经按升序 排列
解题思路
法一:遍历(执行用时:24ms,内存消耗:13MB)
题目已经明确链表是已排序的,所以只要比较相邻的两个节点的值是否相等即可。对链表进行遍历,若当前节点与下一个节点的值相等,则将当前节点指向其下一个节点的下一个节点,否则继续往下比较。
注意,当链表为空或者只有一个节点时,直接返回链表即可。当我们遍历到链表的最后一个节点时,pre.next 为空节点,如果不加以判断,访问 pre.next对应的元素会产生运行错误。因此我们只需要遍历到链表的最后一个节点,而不需要遍历完整个链表。
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def deleteDuplicates(self, head):
"""
:type head: ListNode
:rtype: ListNode
"""
pre = head
if not (head and head.next):
return head
while pre.next:
if pre.val == pre.next.val:
pre.next = pre.next.next
else:
pre = pre.next
return head
时间复杂度:O(n),其中n是链表的长度
空间复杂度:O(1)。
法二:快慢指针/双指针(执行用时:24ms,内存消耗:12.8MB)
定义两个指针slow和fast,并初始化为头节点。当链表没有节点或只有一个节点时直接返回链表。以fast不为空作为循环条件,当两个指针所指节点的值不相等时,slow指针指向fast(删除相等的元素),并将slow指针往后移;不管两个指针所指节点的值相不相等,fast指针都是要往后移动的。也就是说,控制slow指针每次只指向后面与当前节点的值不相等的节点。
注意,遍历完成后,slow.next = None,因为链表中有可能最后几个元素都是相等的,此时只有fast在移动,slow指针的指向还未改变,如果不使它指向空节点,最后输出的结果是没有删除最后一组相同元素的。
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def deleteDuplicates(self, head):
"""
:type head: ListNode
:rtype: ListNode
"""
if not (head and head.next):
return head
slow,fast = head,head
while fast:
if slow.val != fast.val:
slow.next = fast
slow = slow.next
fast = fast.next
slow.next = None
return head