https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/是题目描述
最开始想到的是使用双重for循环
def removeDuplicates(nums):
"""
:type nums: List[int]
:rtype: int
"""
for i in range(len(nums) - 1):
first = nums[i]
for j in range(i + 1, len(nums)):
if nums[j] == first:
del nums[j]
j = j - 1
return nums
这段代码是错误的,因为range()函数里的值一旦变换,i或者j就会重新从range()函数里的初始值开始,总之各种混乱,出错。总结:当判断条件不断更新时,不建议使用range()或者for循环,建议使用while 循环。
修改代码:
def removeDuplicates(nums):
"""
:type nums: List[int]
:rtype: int
"""
i = 0
while (i < len(nums) - 1):
first = nums[i]
j = i + 1
while j < len(nums):
if nums[j] == first:
del nums[j]
j = j - 1
j = j + 1
i = i + 1
count = len(nums)
return count
removeDuplicates([0, 0, 1, 1, 1, 2, 2, 3, 3, 4])
代码能够运行成功,但是在力扣提交后,代码运行超出时间范围,看来是算法的时间复杂度太高,而且这段代码可以对未排序内容进行删除重复项(不看题目的后果),需继续改进。究其原因,是因为在正向遍历过程中,每次删除数组元素会引发大量的数据迁移操作,所以可以逆向排序,逆遍历可以防止删除某个元素后影响下一步索引的定位,改进后的代码
def removeDuplicates_2(nums):
for i in range(len(nums) - 1, 0, -1):
if nums[i] == nums[i - 1]:
del nums[i]
print(nums)
return len(nums)
removeDuplicates_2([0, 0, 1, 1, 1, 2, 2, 3, 3, 4])
时间效率空间效率大大提高,而标准解法是使用双指针
ef removeDuplicates_3(nums):
i = 0
for j in range(1, len(nums)):
if nums[j] != nums[i]:
nums[i + 1] = nums[j]
i += 1
print(nums[0 : i + 1])
return i + 1
removeDuplicates_3([0, 0, 1, 1, 1, 2, 2, 3, 3, 4])
在这种解法中,nums列表并没有删除重复元素,nums[0: i + 1]才是没有重复元素的列表。