leetcode第26题,数组去重,返回不重复的元素个数即可。
注意,这里有一个很重要的先决条件,那就是数组是有序的。
数组内的东西可以随便改,而且超出规定长度的元素不理会,这就给了我们提示,也就是说所有的重复元素都可以提到数组前面就可以了。这样就是原地修改了。
这里采用双指针的办法,代码中i作为游标指针,start作为第二个指针,start一直指向的是经过置换之后不重复数组的末尾,所以遍历的时候,如果发现相邻数字不等的情况,start自增,指向的是后面重复数字区域的第一个,再把不相等的数字换到这个位置,此时start又变成了不重复数字的最后一个了。如果相等,游标指针i会一直自增。
数组有序保证了上面的算法是正确的,因为只有有序,相邻之间比较才有意义,重复的都会聚集在一起。
边界条件,当数组为空数组的时候,循环无法执行。
这里有一个处理原地修改分割数组前后两部分比较常见的思路,就是前后双指针,游标指针负责的是遍历,凡是不满足条件的通通放过(游标指针自增),因为这些都是要放到后面的可以不管。满足条件的需要处理,把他们换到前面去。第二个指针始终指向的是前半部满足条件的数组的最后一个元素。
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
if n == 0:
return 0
counter = 1
start = 0
for i in range(n):
if i != n-1:
if nums[i] != nums[i+1]:
counter += 1
start += 1
nums[start] = nums[i+1]
return counter