【LeetCode】80. Remove Duplicates from Sorted Array II解法及注释

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;
}
};