一、题目:
给你一个 非严格递增排列 的数组 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
已按 非严格递增 排列
int removeDuplicates(int* nums, int numsSize) {
}
二、题解:
我们可以使用双指针的方法。
- 首先,检查数组的大小,如果为 0,则返回 0。
- 初始化一个指针
k
为 1,因为我们至少有一个唯一的元素。 - 遍历数组,从第二个元素开始,比较当前元素和前一个元素:如果两个元素不同,就意味着找到了一个新的唯一元素,我们将其放到
k
指向的位置,并将k
增加 1。 - 循环结束后,返回
k
,这就是去重后的数组中唯一元素的个数。
int removeDuplicates(int* nums, int numsSize) {
if (numsSize == 0) {
return 0; // 如果数组为空,返回 0
}
int k = 1; // k 初始化为 1,因为至少存在一个唯一元素
for (int i = 1; i < numsSize; i++) {
// 只要当前元素与前一个元素不同,就将其放置在 k 的位置
if (nums[i] != nums[k - 1]) {
nums[k] = nums[i];
k++; // 增加 k 的大小
}
}
return k; // 返回唯一元素的个数
}