给你一个有序数组 nums ,请你原地删除重复出现的元素,使每个元素只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,2] 输出:2, nums = [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]
解释:函数应该返回新的长度5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
我们来分析一下思路,以示例2为例:
(1)我们定义3个整型变量,dst、p1、p2,最开始时,dst=p1=p2=0,如下图所示
(2)当nums[p2] = nums[p1]时,p2++;直到nums[p2] != nums[p1],然后nums[dst] = nums[p1],dst++,p1 = p2;p2++,如下图
(3)重复(2)的操作,当nums[p2] != nums[p1]时,然后nums[dst] = nums[p1],dst++,p1 = p2, p2++,如下图
(4)继续重复(2)的步骤
当p2>数组元素个数时,循环结束,然后仍要把nums[p1]的值赋值给nums[dst],dst++;那么此时dst的值就是去重后数组元素的个数。
我们可以发现,思路是把所有去重后的元素从数组nums的开始位置依次向后存放,p2的作用是找到每个元素(去重后)在数组中第一次出现的位置;然后把这个位置给p1;p1把值给dst;dst作用是标记去重后的元素位置
下面我们来实现代码:
int removeDuplicates(int* nums, int numsSize)
{
if(numsSize == 0)
{
return 0;
}
int p1 = 0;
int p2 = 0;
int dst = 0;
while(p2 < numsSize)
{
if(nums[p2] == nums[p1])
{
p2++;
}
else
{
//找到了下一个新元素第一次出现的位置
//把当前元素的值赋值给dst位置
nums[dst] = nums[p1];
//p1是p2找到的下一个新元素位置
p1 = p2;
//p2继续向后寻找下一个新元素第一次出现的位置
p2++;
//dst是下一个新元素应该存放的位置
dst++;
}
}
nums[dst] = nums[p1];
dst++;
return dst;
}
int removeDuplicates(int* nums, int numsSize)
{
if (numsSize == 0)
{
return 0;
}
int p1= 0 , p2 = 0, dst = 0;
while (p2 < numsSize)
{
if (nums[p2] != nums[p1])
{
p1 = p2;
dst++;
nums[dst] = nums[p1];
p2++;
}
else
{
p2++;
}
}
return dst + 1;
}
对于这道题目,也可以选择双指针(下标)
使用下标p1和p2
当nums[p1] == nums[p2]时,p2++,
当nums[p1] != nums[p2]时,p2++, nums[p2] = nums[p1],p1++
int removeDuplicates(int* nums, int numsSize)
{
if (numsSize == 0)
{
return 0;
}
int p1 = 0, p2 = 0;
while (p1 < numsSize)
{
if (nums[p2] == nums[p1])
{
p1++;
}
else
{
p2++;
nums[p2] = nums[p1];
p1++;
}
}
p2++;
return p2;
}