目录
- 标签(题目类型):数组、双指针
题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。
你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”传递的。也就是说,不对实参作任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
思路及实现
方式一:双指针法
思路
利用双指针法,一个指针用于遍历数组,另一个指针指向不重复序列的末尾,即下一个要插入不重复元素的位置。遍历数组时,如果当前元素与前一个元素(即不重复序列的末尾元素)不同,则将该元素插入不重复序列末尾,并移动末尾指针。
代码实现
Java版本
public int removeDuplicates(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
int i = 0; // 慢指针,指向不重复序列的末尾
for (int j = 1; j < nums.length; j++) { // 快指针,遍历数组
if (nums[j] != nums[i]) {
i++;
nums[i] = nums[j];
}
}
return i + 1; // 返回不重复序列的长度
}
说明:
- 如果数组为空或长度为0,直接返回0。
- 初始化慢指针
i
为0,指向不重复序列的第一个元素。- 遍历数组,如果当前元素
nums[j]
与慢指针指向的元素nums[i]
不同,说明找到了一个新的不重复元素,将其插入不重复序列末尾,并移动慢指针i
。- 最后返回
i + 1
,因为i
是指向不重复序列末尾的下一个位置。
C语言版本
int removeDuplicates(int* nums, int numsSize) {
if (nums == NULL || numsSize == 0) {
return 0;
}
int i = 0; // 慢指针,指向不重复序列的末尾
for (int j = 1; j < numsSize; j++) { // 快指针,遍历数组
if (nums[j] != nums[i]) {
i++;
nums[i] = nums[j];
}
}
return i + 1; // 返回不重复序列的长度
}
说明:
- 与Java版本逻辑相同,只是语法有所不同。
Python3版本
def removeDuplicates(nums):
if not nums:
return 0
i = 0 # 慢指针,指向不重复序列的末尾
for j in range(1, len(nums)): # 快指针,遍历数组
if nums[j] != nums[i]:
i += 1
nums[i] = nums[j]
return i + 1 # 返回不重复序列的长度
说明:
- Python版本与Java和C版本逻辑相同,只是语法和习惯用法
有所不同。
复杂度分析
- 时间复杂度:O(n),其中n是数组的长度。因为每个元素只被遍历一次。
- 空间复杂度:O(1)。只使用了常数级别的额外空间。
方式二:使用集合(不适用于原地修改要求)
思路
将数组中的元素放入一个集合中,自动去重,然后再将集合中的元素放回数组。但这种方式不满足题目要求的原地修改和O(1)额外空间的要求,因此不推荐在实际题目中使用。
代码实现
Java版本
import java.util.LinkedHashSet;
import java.util.Set;
public int removeDuplicates(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}
Set<Integer> set = new LinkedHashSet<>();
for (int num : nums) {
set.add(num);
}
int index = 0;
for (int num : set) {
nums[index++] = num;
}
return set.size();
}
说明:
- 使用了
LinkedHashSet
来保持元素的插入顺序。- 遍历数组,将元素加入集合中自动去重。
- 遍历集合,将元素放回数组。
- 返回集合的大小,即去重后的数组长度。
C语言版本
由于C语言没有内置集合类型,实现起来相对复杂,需要手动实现哈希表或链表等数据结构,这里不给出C语言的集合实现方式。
Python3版本
def removeDuplicates(nums):
if not nums:
return 0
nums[:] = list(dict.fromkeys(nums))
return len(nums)
说明:
- 利用Python的字典
dict
来自动去重,dict.fromkeys(nums)
会创建一个以nums
为键的字典,由于字典的键是唯一的,所以会自动去重。- 然后将字典的键转换回列表,并赋值给原数组
nums
。- 返回列表的长度,即去重后的数组长度。
复杂度分析
- 时间复杂度:O(n),其中n是数组的长度。遍历数组和集合操作都是线性的。
- 空间复杂度:O(n)。使用了额外的集合来存储去重后的元素。
总结
方式 | 优点 | 缺点 | 时间复杂度 | 空间复杂度 |
---|---|---|---|---|
方式一 | 原地修改,满足题目要求 | 代码稍微复杂一些 | O(n) | O(1) |
方式二 | 代码简洁,易理解 | 不满足原地修改和O(1)额外空间的要求 | O(n) | O(n) |
相似题目
相似题目 | 难度 | 链接 |
---|---|---|
27. 移除元素 | 简单 | 力扣-27 |
80. 删除有序数组中的重复项 II | 中等 | 力扣-80 |
268. 缺失数字 | 简单 | 力扣-268 |
这些题目都与数组操作、去重和移除元素有关,可以通过练习这些题目来加深对数组操作的理解。
欢迎一键三连(关注+点赞+收藏),技术的路上一起加油!!!代码改变世界
关于我:阿里非典型程序员一枚 ,记录在大厂的打怪升级之路。 一起学习Java、大数据、数据结构算法(公众号同名),回复暗号,更能获取学习秘籍和书籍等
—⬇️欢迎关注下面的公众号:
进朱者赤
,认识不一样的技术人。⬇️—