删除排序数组中的重复项(I和 II)

描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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。
你不需要考虑数组中超出新长度后面的元素。

注意在题目当中要求的是:不要使用额外的数组空间,你必须在 原地修改输入数组 并在使用 O(1) 额外空间的条件下完成,意味着不能通过新建数组存放的方式解决,那么对于这样一个问题我们该如何解决呢?

可以考虑增加一个游标的方式(命名为:index),遍历整个数组,当遇到前一个和后一个不相等的时候就给游标自增,这样的话对于相同的元素可以直接跳过不统计,最后返回”游标的大小+1“(也即不重复的元素的个数),代码实现如下:

class Solution
{
public:
    int removeDuplicates(vector<int>& nums)
    {
    if (nums.empty())
        return 0;
        
    int index = 0;
    for (int i = 1; i < nums.size(); i++)
    {
        if (nums[index] != nums[i])
        nums[++index] = nums[i];
    }
    return index + 1;
  }
};

这样的话就符合了题目所要求的各个复杂度以及就地统计的原则,当然,如果你对 STL 比较熟悉的话这道题其实可以偷懒一下:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
    	return distance(nums.begin(), unique(nums.begin(), nums.end()));
    }
};

比较容易是不是?我们对题目做一点改变,由于传感器有一个误差的区间,对于所有可能的种类我们允许它最多出现两次,认为是不同的两个种类,这样的话问题该如何解决呢?

由于要求是两次,所以我们可以考虑通过增加一个变量的方式来记录种类编号出现的次数就可以了嘛,代码实现如下:

class Solution {
    public:
    int removeDuplicates(vector<int>& nums) {
    if (nums.size() <= 2)
        return nums.size();
        
    int index = 2;
    for (int i = 2; i < nums.size(); i++){
        if (nums[i] != nums[index - 2])
        nums[index++] = nums[i];
    }
    return index;
    }
};

链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

全部评论

相关推荐

杨柳哥:这不是普通人,那这个钱的是天才
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务