26. 删除排序数组中的重复项
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。
你不需要考虑数组中超出新长度后面的元素。
思路误区:
- 一直以为只要输出长度正确,就能返回正确的值。
- 没有想要用指针,同时总想着从前到后走,所以在使用python中list的pop时,总会出现超出index的错误。
JAVA
思路:
- 用指针的思想。
- 要将不同的值移到数组的左侧。
- 数组是有序的,那么重复的元素一定会相邻。
- 考虑用 2 个指针,一个在前记作 p,一个在后记作 q,算法流程如下:
比较 p 和 q 位置的元素是否相等。
如果相等,q 后移 1 位
如果不相等,将 q 位置的元素复制到 p+1 位置上,p 后移一位,q 后移 1 位
重复上述过程,直到 q 等于数组长度。
返回 p + 1,即为新数组长度。
class Solution { // 用指针
public int removeDuplicates(int[] nums) {
if (nums.length == 0 ||nums ==null)return 0;
int p = 0;
int q = 1;
while(q<nums.length){
if(nums[p]==nums[q]){
q++;
}else{
nums[p+1] = nums[q];
p++;
q++;
}
}
return p+1;
}
}
python
思路:
- 从后往前看数组,如果和前一个相同则删除。
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
for i in range(len(nums)-1,0,-1):
if nums[i] == nums[i-1]:
nums.pop(i)
return len(nums)