问题描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。 你不需要考虑数组中超出新长度后面的元素。
解题方法
刚开始我用的是一个指针,每扫描到一个元素判断是否与后一个元素相等,如果相等就将它移到数组最后,但是这个方法太慢了
看到了大佬的双指针方法,
-
用 2 个指针,一个在前记作 p,一个在后记作 s 比较 p 和 s位置的元素是否相等。 如果相等,s 后移 1 位 如果不相等,将 s 位置的元素覆盖给 p+1 指向的元素,然后p后移一位,s后移 1 位 重复上述过程,直到走完数组
过程如下:
覆盖之后p指针和s指针后移:
重复此过程,走完后不重复的值都被放在了数组的左侧
代码
public static int removeDuplicates1(int[] nums) { //单指针
if(nums.length== 1)
return nums.length;
int i = 0;
int curLength = nums.length;
while(i < curLength - 1) {
if(nums[i + 1] == nums[i]) {
for (int j = i + 1; j < curLength - 1; j++) {
int temp = nums[j];
nums[j] = nums[j + 1];
nums[j + 1] = temp;
}
curLength--;
}else {
i++;
}
}
return curLength;
}
public static int removeDuplicates2(int[] nums) { //双指针法
int pre = 0; //前指针
int suc = 1;//后指针
while(suc < nums.length) {
if(nums[suc] == nums[pre]) {
suc++;
}else {
nums[pre + 1] = nums[suc];
pre++;
suc++;
}
}
return pre + 1;
}