80. Remove Duplicates from Sorted Array II
Total Accepted: 71844 Total Submissions: 220521 Difficulty: Medium
Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?
For example,
Given sorted array nums = [1,1,1,2,2,3]
,
Your function should return length = 5
, with the first five elements of nums being 1
, 1
, 2
, 2
and 3
. It doesn't matter what you leave beyond the new length.
【分析】
此题的关键在于不利用额外空间而完成数字的删除,因此,我们只能在现有空间,也就是输入数据所在的空间对数据进行操作。根据题意,我们需要将去除重复后的数据依次排在原数组的前面,因此,不可免要对数据进行移动操作,关于移动数据,一般有两种方案:1、整体移动,当遍历到题目所指的重复数据后,我们将其后面的数据整体移动一位,从而将重复数据覆盖,如此,最糟糕的情况,输入的数组n个数据全相同,我们需要移动[(n-3)+(n-4)+...+0],当n比较大时,时间复杂度为O(n2);2、“双指针”,一个指针负责遍历数组,一个指针始终指向下一个无重复数据的存放位置,逐次移动,时间复杂度为O(n).
【解法及注释】
根据分析,我将两种方法的程序均进行了测试,效率相差不多,这可能与测试案例的数据规模比较有关。
方法一:
class Solution33 {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size()<3)return nums.size();
int result=1;//记录删除重复后的数据长度
int count=0;//标记重复
int move=0;//移动次数
for(int i=0;i<nums.size()-1;)
{
if(nums[i]>nums[i+1])//原数组是排序的,每次遇到重复,就将其移动到数组的末尾,其后的数据则前移1位
{ //因此,数组尾部的数据会比原数组尾部的数据小,这个特点可以作为判断遍历完成的条件
break;
}
else if(nums[i]!=nums[i+1])//相邻元素不等
{
result++;
count=0;
i++;//控制循环
move=0;
}
else if(nums[i]==nums[i+1])//相邻数据相等
{
if(count==2)//已经重复(超过2)
{
count=2;
move++;//记录移动次数
int temp=nums[i+1];//把重复数据移动到数组尾部
for(int k=i+1;k<nums.size()-1;k++)//整体前移
{
nums[k]=nums[k+1];
}
nums[nums.size()-1]=temp;//交换到尾部
if(move>nums.size()-i-2)//避免陷入死循环
i++;
}
else //该数据两次出现
{
count=2;//标记
result++;//记录
i++;//进入下一次循环
move=0;
}
}
}
return result;
}
};
方法二:
class Solution {
public:
int removeDuplicates(vector<int> &nums)
{
if(nums.size() < 3) return nums.size();
int length = 1;
int flag = 1;//标记重复(超过2次)
for(int i = 1; i < nums.size(); i++)
{
if(nums[i] != nums[i-1])//相邻数据不等,将数据存储到length指向的空间
{
nums[length++] = nums[i];
flag = 1;//标记
}
else if(flag == 1)//出现相等数据,且与之前的数据不等
{
nums[length++] = nums[i];//将数据存储到length指向的空间
flag++;//标记改变
}
else//出现相等数据,且之前已经出现了2个,因此i指向的数据不是期望的数据,不保存
continue;
}
return length;
}
};