LeetCode的第26题是非常简单的一道题目,大意就是给定一个排序好的数组,但是该数组中存在重复元素,要求操作该数组,去掉其中的重复元素,同时统计去重之后数组的长度。原题的做法是使用两个指针,比较对应元素,同时修改数组,附上O(n)Python代码:
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
index = 0
length = len(nums)
if length < 2:
return length
for i in range(1,length):
if nums[index] != nums[i]:
index += 1
nums[index] = nums[i]
index += 1
return index
但是,解决了这个题目之后,我们还可以思考另一个问题,本题麻烦的地方在于要求修改原数组,但方便的地方是给定的数组是排序过的,那现在我们修改一下原题的条件,首先不需要修改原数组,但是数组元素变成无序的,这时候如何解决呢?使用Python里的字典同样可以实现O(n)的效率。代码附上:
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
result = 0
cnt = {}
for i in range(len(nums)):
if nums[i] not in cnt:
cnt[nums[i]] = i
return len(cnt)
遍历一遍数组,只要不在字典中的就把该元素添加进去,但是对于重复元素,由于字典中已经存在,则不会添加,最后得到的字典的长度就是不重复的元素的个数