【LeetCode-数组】删除排序数组中的重复项

六眼飞鱼酱① 提交于 2020-04-06 23:59:54

题目描述

给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例:

给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。

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

思路

做这题之前可以先做一下移除元素。使用一快一慢两个指针,i为慢指针,j为快指针,都从数组头开始遍历。如果nums[i]==nums[j],则j++,继续遍历下一个元素。如果nums[i]!=nums[j],则说明找到了不同的元素,因为重复的元素要保留一个,所以先将i+1后赋值,也就是nums[++i]=nums[j],当j遍历完毕时,当前数组的长度为i+1.代码如下:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.empty())
            return 0;
        
        int i=0, j=0;
        int curLen = nums.size();   // 当前数组长度
        while(j<curLen){
            if(nums[i]!=nums[j]){
                i++;
                nums[i]=nums[j];
                
            }
            j++;
        }
        return i+1;
    }
};
  • 时间复杂度O(n)
    j遍历一遍即可,所以O(n).
  • 空间复杂度O(1)
    申请的额外内存不随输入规模的增大而增大。

相似题目

移除元素:https://leetcode-cn.com/problems/remove-element/

总结

当遇到移除元素的题目时,可以尝试使用一快一慢两个指针来解决。快指针慢指针起始位置相同,当快指针指向的元素与慢指针指向的元素不同时进行相应的操作。

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!