题目说明
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/
解法1
时间复杂度:O(n)
空间复杂度:O(1)
思路:使用双指针,第一个指针j表示调整后的数组最后一个位置,第二个指针i遍历原数组。
当nums[i]与nums[j]不等时,直接添加即可,将i位置添加到新数组末尾,并记录下匹配次数(此时为1)。
当nums[i]与nums[j]相等时,就需要判断匹配次数了,判断是否满足匹配次数小于2,若满足则将位置添加到新数组末尾,匹配次数加1;不满足,则直接跳过。
若要修改最大出现的次数,修改max值即可。
int removeDuplicates(vector<int>& nums)
{
if (nums.empty())
return 0;
int j = 0;
int count = 1;
int max = 2;
for (int i = 1; i < nums.size(); i ++)
{
if (nums[i] == nums[j] && count < max)
{
nums[++j] = nums[i];
count ++;
}
else if (nums[i] != nums[j])
{
nums[++j] = nums[i];
count = 1;
}
}
return j + 1;
}