题目来源
Remove Duplicates from Sorted Array II
问题描述
“删除重复项目” 的进阶:
如果重复最多被允许两次,又该怎么办呢?
例如:
给定排序数列 nums = [1,1,1,2,2,3]
你的函数应该返回长度为 5,nums 的前五个元素是 1, 1, 2, 2 和 3。
解决方案
C++
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int k = 1;//只允许两次
int size = nums.size();
if (size == 0 || size == 1)
return size;
int pos = 0;
for (int i = 1; i < size; i++)
{
if (nums[i] != nums[pos])
{
k=1;
pos++;
nums[pos] = nums[i];
}
else if (k<2){
k++;
pos++;
nums[pos] = nums[i];
}
}
return pos+1;
}
};
Java
class Solution {
public int removeDuplicates(int[] nums) {
if (nums.length <= 2)
return nums.length;
int index = 2;
for (int i = 2; i < nums.length; i++) { //从第三个元素开始
if (nums[i] != nums[index - 2]) //由于排好了序,相等说明 i至少是第三个重复元素
{
nums[index] = nums[i];
index++;
}
}
return