LeetCode 80. Remove Duplicates from Sorted Array II

欢迎访问原文所在博客：52Heartz’s Blog

解答1[Java]：

class Solution {
    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(); // if array size is less than k then return the same
    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 位。