题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成
示例1
给定数组 nums = [1,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。
你不需要考虑数组中超出新长度后面的元素。
方法一:暴力解法
思路
老规矩,先来一遍暴力解法。与 27 题移除元素类似,这道题的暴力解法是两层 for 循环,第一层遍历数组,第二层更新数组,如果要删除某个元素,则该元素后面的所有元素都要进行前移操作
代码
class Solution {
public int removeDuplicates(int[] nums) {
int length = nums.length;
for (int i = 1; i < length; ++i) {
if (nums[i] == nums[i - 1]) {
for (int j = i - 1; j < length - 1; ++j) {
nums[j] = nums[j + 1];
}
--length;
--i; // 注意 --i 要放在 if 语句中,只有 if 语句成立的时候才需要进行 i--
}
}
return length;
}
}
复杂度分析
时间复杂度:O(n^2)
空间复杂度:O(1)
方法二:双指针法
思路
保留两个 i 和 j,i 是满指针,j 是快指针,当 nums[j] 与给 nums[i] 相等时,递增 j 以跳过该元素,否则就将 nums[j] 复制到 nums[i],并同时递增两个指针。重复这一过程,直到 j 到达数组的末尾。此时数组的长度为 i。
代码
class Solution {
public int removeDuplicates(int[] nums) {
int i = 0;
int length = nums.length;
for (int j = i + 1; j < length; ++j) {
if (nums[j] != nums[i]) {
++i;
nums[i] = nums[j];
}
}
return i + 1;
}
}
复杂度分析
时间复杂度:O(n), 假设数组的长度是 n,那么 i 和 j 分别最多遍历 n 步。
空间复杂度:O(1)。