26. 删除有序数组中的重复项
1.题目介绍
给你一个 非严格递增排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。
考虑 nums 的唯一元素的数量为 k ,你需要做以下事情确保你的题解可以被通过:
更改数组 nums ,使 nums 的前 k 个元素包含唯一元素,并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。返回 k 。
示例 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 已按 非严格递增 排列
2.题解
2.1 哈希表
看到出现次数类似的问题,没有多想,使用哈希表记录次数进行判断是肯定可以的,如下
代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
unordered_map<int,int> map;
int count = 0;
for (int i = 0; i < nums.size(); i++){
int num = nums[i];
if (!map[num]) nums[count++] = num;
map[num]++;
}
return count;
}
};
2.2 双指针(快慢指针)
草率了,没有看清题目nums 已按非严格递增排列(只看到个非字QAQ),所以这里既然已经排序过了,那么使用快慢指针便是很好的方法
思路
快指针指向旧数组被遍历的元素,慢指针指向新数组(借着旧数组的壳子)的最新一个元素(非重复),只有当检测到非重复元素,快慢指针一起移动;否则只有快指针移动
- 首先注意到这里
1 <= nums.length <= 3 * 104
,说明数组nums至少有一个元素,所以就不用考虑数组为空的情况了! - 这里第一个元素肯定是不会重复的,所以就以其为开始基准,后面开始判断
- 这里的
slow
虽然是下标,但最后的slow++
多加了一个,正好返回的就是长度了
代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() == 1) return 1;
int fast = 1, slow = 1, pre = nums[0];
for (; fast < nums.size(); fast++){
if (nums[fast] != pre){
nums[slow++] = nums[fast];
pre = nums[fast];
}
}
return slow;
}
};
2.3 代码优化
思路
考虑如下数组:
此时数组中没有重复元素,按照上面的方法,每次比较时 nums[p] 都不等于 nums[q],因此就会将 q 指向的元素原地复制一遍,这个操作其实是不必要的。
因此我们可以添加一个小判断,当 q - p > 1 时,才进行复制。
作者:Max
链接: [https://leetcode.cn/problems/remove-duplicates-from-sorted-array/solutions/34033/shuang-zhi-zhen-shan-chu-zhong-fu-xiang-dai-you-hu/
来源:力扣(LeetCode)]
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码
这里在2.2代码基础上还同时优化掉了pre,直接使slow从0开始(这里slow是下标,所以最后返回长度的时候,不要忘了返回slow+1)。
这里这个slow++的位置很关键,请不要放在 if(fast - slow > 1)判断前面,会影响到判断二者间距。同时也不能使用 nums[++slow] = nums[fast];·这样若是出现上图所示情况,会无法进入该判断,而导致slow一直无法推进。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() == 1) return 1;
int fast = 1, slow = 0;
for (; fast < nums.size(); fast++){
if (nums[fast] != nums[slow]){
if(fast - slow > 1) nums[slow + 1] = nums[fast];
slow++;
}
}
return slow + 1;
}
};
2.4 思考
思路
能不能像 27.移除元素 那样使用首尾指针的方法?
注意题目要求:元素的相对顺序应该保持一致 。但是首尾指针的方法会打乱数组元素的相对顺序,所以排除。