Leetcode-删除有序数组中的重复项(中等)

删除有序数组中的重复项

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

输入:nums = [1, 1, 2]
输出:2, nums = [1, 2]

快慢指针分别在索引为0和1的位置,慢指针时钟指向不重复序列的最后一位,快指针则否则遍历
遇到与慢指针相同时就继续++,不同时则慢指针后移,将下一个不重复的数字复制过来

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        int low = 0, fast = 1;
        while (fast < nums.size()) {
            if (nums[low] != nums[fast]) {//不相等时,快指针复制到慢指针的下一位;相等时只有快指针+1
                low++;
                nums[low] = nums[fast];
            }
            fast++;
        }
        return low + 1;
    }
};

时间复杂度:O(n),假设数组的长度是 n,那么 i 和 j 分别最多遍历 n步。
空间复杂度:O(1)

升级:
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 最多出现k次 ,返回删除后数组的新长度。

//------------------------------------way1
class Solution {
public:
    //定义通用函数,使每个元素最多出现k次 
    int remove(vector<int>& nums, int k) {
        if (nums.size() <= k) return nums.size();
        int low = k, fast = k;//low始终指向已排好的新数组的下一位
        while (fast < nums.size()) {
            if (nums[low - k] != nums[fast]) {  // k=2时候  3(low-2) 3 xlow 3(fast) 视为重复,此时low - k=fast
                nums[low] = nums[fast];
                low++;
            }
            fast++;
        }
        return low;
    }
    //------------------------------------way2
public:
    //定义通用函数,使每个元素最多出现k次 
    int remove(vector<int>& nums, int k) {
        if (nums.size() <= k) return nums.size();
        int low = k - 1, fast = k;//low始终指向已排好的新数组的最后一位
        while (fast < nums.size()) {
            if (nums[low - k + 1] != nums[fast]) {  // k=2时候  3(low-1) 3(low) 3(fast) 视为重复,此时 low - k + 1=fast
                low++;
                nums[low] = nums[fast];
            }
            fast++;
        }
        return low + 1;
    }
全部评论

相关推荐

头像
11-09 17:30
门头沟学院 Java
TYUT太摆金星:我也是,好几个华为的社招找我了
点赞 评论 收藏
分享
AFBUFYGRFHJLP:直接去美帝试试看全奖phd吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务