给你一个 非严格递增排列 的数组 nums
,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
考虑 nums
的唯一元素的数量为 k
,你需要做以下事情确保你的题解可以被通过:
- 更改数组
nums
,使nums
的前k
个元素包含唯一元素,并按照它们最初在nums
中出现的顺序排列。nums
的其余元素与nums
的大小不重要。 - 返回
k
。
判题标准:
系统会用下面的代码来测试你的题解:
int[] nums = [...]; // 输入数组 int[] expectedNums = [...]; // 长度正确的期望答案 int k = removeDuplicates(nums); // 调用 assert k == expectedNums.length; for (int i = 0; i < k; i++) { assert nums[i] == expectedNums[i]; }
如果所有断言都通过,那么您的题解将被 通过。
示例 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
。不需要考虑数组中超出新长度后面的元素。
提示:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums
已按 非严格递增 排列
代码的主要思路是使用两个指针fast
和slow
来遍历数组并移除重复元素。fast
指针每次移动一步,而slow
指针仅在找到非重复元素时移动。
下面是对代码的逐步分解:
-
初始化
n
为输入数组nums
的大小,并将fast
和slow
指针初始化为1,因为我们假设数组的第一个元素不是重复的。 -
检查输入数组是否为空。如果为空,则返回0,因为没有需要移除重复元素的元素。
-
开始一个while循环,循环条件是
fast
指针没有到达数组的末尾。 -
在循环内部,检查当前
fast
指针处的元素是否不等于前一个元素(即找到了一个非重复元素)。 -
如果当前元素不是重复的,将
slow
位置上的元素更新为当前元素,并将slow
指针加1。 -
循环结束后,返回
slow
指针的值,该值表示数组中非重复元素的个数。
该解决方案的时间复杂度是O(n),其中n是输入数组的大小。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int n=nums.size(),fast=1,slow=1;
if(n==0)
return 0;
while(fast<n){
if(nums[fast]!=nums[fast-1]){
nums[slow]=nums[fast];
++slow;
}
++fast;
}
return slow;
}
};