题目
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。
你不需要考虑数组中超出新长度后面的元素。
思路
解决数组问题的一个有效的方法就是利用双指针
,在这题里面,因为是已经排好序的,可以让i
指针指向插入的位置,j
指针找不同数字的位置就可以了,所以可以采取如下步骤:
- 定义两个指针
i
,j
,一个在前,一个在后,初始时都指向首位 j
指向的数字与i
指向的数字就往后迭代,直到不相同i
指针判断下一个位置是否是j
指针所在位置,是,则i
往后移一位,否则将i+1
所指向位置的值变为j
指向的数字,再i
往后移一位- 重复以上步骤,直到
j
超过数组的长度
public int removeDuplicates(int[] nums) {
if(nums == null){
return 0;
}
int i=0,j=0;
int count=0;
while (j<nums.length){
while (j < nums.length && nums[i] == nums[j]){
j++;
}
if(i+1 == j){
i++;
}else if (j < nums.length){
nums[i+1]=nums[j];
i++;
}
count++;
}
return count;
}