Leetcode算法——26、有序数组去重

给定一个有序数组，去除重复的元素，返回新的数组长度。
备注：
如果返回的新数组长度为m，则需要保证数组的前m个元素一定包含了所有的去重后的元素，之后的元素值不作要求。
不要为另外的数组去开辟新的空间，只能修改原数组，空间复杂度为O(1)。

示例：
输入： nums = [1,1,2],
输出：2，且保证 nums 的前2位为唯一元素。

思路

由于是有序数组，因此所有重复元素肯定都是相邻的。
因此判断一个元素是否重复，只需要观察它的前一个元素是否与它相同即可。

判断出一个重复元素后，如何处理它，有两种选择：

• 从左到右扫描，维护一个变量，始终等于前一个元素。这样就可以知道当前元素是否是重复元素。
  如果重复了，删除这个元素。
  此方法会改变原数组的长度。
• 从左到右扫描，维护一个索引，这个索引始终保持为去重后元素的最后一位。
  如果扫描到了一个元素和这个索引的元素不同，则将索引+1，且索引对应的值修改为当前元素值。
  此方法不会改变数组的长度。

两种方法均只需要扫描一遍，因此时间复杂度均为 O(n)。

python实现

def removeDuplicates(nums):
    """
    :type nums: List[int]
    :rtype: int
    如果重复了，删除这个元素。
    """
    if len(nums) <= 1:
        return len(nums)
    
    idx = 1
    while(idx < len(nums)):
        if nums[idx] == nums[idx - 1]:
            del nums[idx]
        else:
            idx += 1
    return idx

def removeDuplicates2(nums):
    """
    :type nums: List[int]
    :rtype: int
    移动重复元素。
    """
    if len(nums) <= 1:
        return len(nums)
    
    idx = 0
    for num in nums:
        if num != nums[idx]:
            idx += 1
            nums[idx] = num
    return idx + 1

if '__main__' == __name__:
    nums = [1,1,2]
    print(removeDuplicates2(nums))