题目
给你一个 升序排列 的数组 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 已按 升序 排列
思路
有序数组原地去重,是指在不使用额外空间(即空间复杂度为O(1))的情况下,对有序数组进行去重操作。具体实现可以采用双指针的方式,将重复的元素移动到数组末尾,最终返回去重后的数组长度。具体实现如下:
function removeDuplicates(nums) {
if (nums.length === 0) return 0;
let i = 0;
for (let j = 1; j < nums.length; j++) {
if (nums[j] !== nums[i]) {
i++;
nums[i] = nums[j];
}
}
return i + 1;
}
在这个实现中,我们使用双指针i和j,其中i指向去重后的数组的末尾,j指向当前遍历的元素。如果第j个元素与第i个元素不相等,则将第j个元素赋值给第i+1个元素,并将i指向i+1。如果相等,则说明该元素重复,直接跳过。最终,返回去重后的数组长度i+1。 这种实现方式的时间复杂度为O(n),空间复杂度为O(1)。因为数组是有序的,所以我们可以使用双指针的方式,快速地找到重复的元素,并将其移到数组末尾。由于不使用额外空间,所以空间复杂度为O(1)。 需要注意的是,该实现只能用于有序数组,如果是无序数组,需要使用其他算法。