leetcode26： 删除排序数组中的重复项

Posted 2020-11-20 Kaitlyn

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。 

你不需要考虑数组中超出新长度后面的元素。

示例 2:

给定 nums = [0,0,1,1,1,2,2,3,3,4],

函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。

你不需要考虑数组中超出新长度后面的元素。

说明:

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

 思路：

设置两个索引i和j，一个表示不重复元素的位置，一个表示遍历的元素的位置。因为是排序好的数组，当j处的元素等于i处时，就不做处理，j直接往下一位；当不等于时，把j处的值赋给i+1位。j再往下移一位。

 1 def removeDuplicates(nums):
 2     """
 3     :type nums: List[int]
 4     :rtype: int
 5     """
 6     if  nums == []:  # 处理输入较异常的两种情况：为空和只有一个元素
 7         return 0
 8     elif len(nums) == 1:
 9         return 1
10     else: #  当数组元素大于一个时
11         i = 0  # 记录不重复元素的位置
12         j = i+1  # 从i的下一个位置开始遍历，如果等于i处的元素则不处理，j加一，如果不等于，则把j处的值赋值给i的下一位
13         while j<len(nums):
14             if nums[j] == nums[i]:
15                 j += 1
16             else:
17                 nums[i+1] = nums[j]
18                 i += 1
19                 j += 1
20         return i + 1
21 print(removeDuplicates([1,1,2]))