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

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

题目描述

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 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。
你不需要考虑数组中超出新长度后面的元素。

解题思路

原地修改数组，定义一个变量记录不同元素的下标，循环整个数组，当前元素和前一个元素相同时，不做任何操作；不相同时，记录不同元素的下标累加，并把当前的元素值更新到当前下标下，代码如下：

class Solution {
     
public:
    int removeDuplicates(vector<int>& nums) {
     
        // 当数组中元素为空或者只有一个元素时，直接返回数组的长度
        if(nums.size() <= 1){
     
            return nums.size();
        }
        
        int count = 0; // 记录当前不同元素的下标位置
        for (int i=1; i<nums.size(); i++){
     
            if (nums[i] == nums[count]){
     
                continue;
            }
            else{
     
                nums[++count] = nums[i];  // 把不同的元素更新到当前的位置
            }
        }
        return count+1;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步