小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。
题目描述
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[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 。不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 已按升序排列
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
复制代码
思路 & CODE
1. 暴力循环
根据题目要求,时间复杂度必须为O(1)
,也就是说必须在原数组上操作,不能够开辟另外的空间。
看到这道题第一想法还是暴力循环
- 首先遍历整个数组
- 从第一个元素开始到当前遍历的索引位置再次进行遍历,看看当前索引位置的元素是否与前面的元素重复
- 如果不重复,进入下次循环
- 如果重复,就去当前索引位置之后的元素找一个去当前索引位置之前的元素值不相同的值与当前位置替换,如果找不到就直接返回当前位置的索引
代码如下:
public int removeDuplicates(int[] nums) {
boolean same = false;
for (int i = 0; i < nums.length; i++) {
same = false;
for (int j = 0; j < i; j++) {
if (nums[j] == nums[i]) {
same = true;
}
}
if (same) {
if (i == nums.length - 1) {
if (same) {
return i;
} else {
return i + 1;
}
}
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] != nums[i] && !preHas(i, j, nums)) {
nums[i] = nums[j];
break;
}
if (j == nums.length - 1) {
return i;
}
}
}
}
return nums.length;
}
private boolean preHas(int i, int j, int[] nums) {
for (int m = 0; m < i; m++) {
if (nums[m] == nums[j]) {
return true;
}
}
return false;
}
复制代码
虽然这个算法能跑通,但是时间复杂度是O(n^2),只超过了5%的人
2. 双指针(快慢指针)
根据官方题解,我们来理一下思路
首先题目告诉我们数组是有序的,所以只需要将一个元素与它的前一个元素比较就能确定这个元素是否重复
我们定义两个指针:
扫描二维码关注公众号,回复:
13168396 查看本文章
- 快指针:遍历数组到达的下表位置
- 慢指针:下一个不同元素要填入的位置
根据上面的定义,我们可以得出结论,我们通过操作快指针找到重复元素,然后把这个元素替换到慢指针所在的索引位置,最后快慢指针索引加一。
- 如果数组长度为0,则不包含任何元素,返回0
- 如果数组长度大于0,那么至少返回一个元素,我们直接从索引为1的位置开始删除即可
- 到最后,慢指针的索引就表示当前有多少个不重复元素
代码如下:
public int removeDuplicates(int[] nums) {
int fast = 1;
int slow = 1;
for (int i = 1; i < nums.length; i++) {
if (nums[fast] > nums[fast - 1]) {
nums[slow++] = nums[fast];
}
fast++;
}
return slow;
}
复制代码
看了题解才发现这种解法,自己第一时间还是无法想出来,总结点双指针的规律:
- 数组有序
- 需要对两个元素进行比较