首页 > itarticle > leetcode 80. remove duplicates from sorted array ii [medium] 补充解答：修改版

leetcode 80. remove duplicates from sorted array ii [medium] 补充解答：修改版

admin 11月 13, 2020 0

题目来源：https://leetcode.com/problems/remove-duplicates-from-sorted-array-ii/

题目难度：Medium

class  {
    public int removeDuplicates(int[] nums) {
        int i = 0;
        for (int n : nums)
            if (i < 2 || n > nums[i - 2])
                nums[i++] = n;
        return i;
    }
}

补充解答：修改版

int removeDuplicates(vector<int>& nums,int k) {
    if(nums.size()<k) return nums.size(); 
    int i,j;
     for(i=k,j=k ; i<nums.size();i++)
         if(nums[j-k]!=nums[i]) nums[j++]=nums[i];
     return j;
}

这个解答，可以适用于保留最多不超过 K 个相同的元素。

和第一种解法的思路相似，j-k 就是 lookback，向后看。第一种解法是 nums[i-2]，往后看两位。现在是往后看 k 位。