题目描述:给定一个排序链表,删除所有重复的元素,使得每个元素只出现一次。
示例 1:
输入: 1->1->2
输出: 1->2
示例 2:
输入: 1->1->2->3->3
输出: 1->2->3
当看到题目的时候,我就有一个想法,怎么判别元素是否重复呢,如果把元素值放进list中,然后每个元素都要从list中遍历一遍,时间复杂度是不是太高了一点,所以我的第一想法使用set,每有一个元素添加到set中,如果set的长度不变,那么这个元素就是重复的,所以我的代码如下:
temp=ListNode(None)#添加一个空节点指向head
temp.next=head
if head==None:
return None
if head.next==None:
return head
s=set()
while temp.next:
len_set=len(s)
s.add(temp.next.val)
len2_set=len(s)
if len_set==len2_set:#判别set的长度是否发生变化
temp.next=temp.next.next
#如果变化,删除节点,记住这种情况下temp不需要移动
else:
temp=temp.next
if temp==None:
break
return head
后来,当我码完代码时,我发现这个题目的链表是有顺序的,OK,fine,多此一举,还用set占用了额外的空间。只要判定相邻两个数是否相等就ok了。于是乎,再写了一遍:
temp=head
if head==None:
return None
if head.next==None:
return head
while temp.next:
if temp.val==temp.next.val:
temp.next=temp.next.next
else:
temp=temp.next
if temp==None:
break
return head
不过下次,就算你无序,我也可以解决你,哼哼。