给你一个有序数组 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)
{
int p1 = 0;
int p2 = 0;
int dst = 0;
while(p2 < numsSize)
{
if(nums[p2] == nums[p1])
{
p2++;
}
else
{
nums[dst] = nums[p1];
p1 = p2;
p2++;
dts++;
}
}
nums[dst] = nums[p1];
dst++;
return dst;
}
|