26. 删除有序数组中的重复项

发布时间 2023-10-27 18:17:57作者: DawnTraveler

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.首先注意到这里1 <= nums.length <= 3 * 104,说明数组nums至少有一个元素,所以就不用考虑数组为空的情况了!
2.这里第一个元素肯定是不会重复的,所以就以其为开始基准,后面开始判断
3.这里的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;
    }
};