LeetCode 80. Remove Duplicates from Sorted Array II [Medium]

发表于 | 更新于 | 分类于 ， | 阅读次数：

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

题目难度：Medium

解答1[Java]：

1
2
3
4
5
6
7
8
9
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;
    }
}

补充解答：修改版

1
2
3
4
5
6
7
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 位。