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.
计算去除重复后的array的长度很简单,可以通过一个临时向量来存储去除重复后的array,然后将临时向量赋给array。
代码如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
vector<int> temp;
int count = 0;
for(int i=0;i<nums.size();i++)
{
int tick = 0;
if(i != nums.size()-1 && nums[i] == nums[i+1])
{
while(nums[i] == nums[i+1])
{
tick++;
if(tick<3)
temp.push_back(nums[i]);
i++;
}
tick++;
if(tick<3)
temp.push_back(nums[i]);
}
else
{
temp.push_back(nums[i]);
}
if(tick >= 2)
count += 2;
else
count ++;
}
nums = temp;
return count;
}
};
感觉自己写的代码像一坨shit,跟简洁优雅八竿子打不着,需要进行改进。然后就是如何不使用临时变量来实现功能?
代码如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int count = 0;
int start=0;
int len = nums.size();
for(int i=0;i<nums.size();i++)
{
int tick = 0;
if(i != nums.size()-1 && nums[i] == nums[i+1])
{
while(nums[i] == nums[i+1])
{
tick++;
if(tick<3)
{
nums[count++] = nums[i];
}
i++;
}
tick++;
if(tick<3)
{
nums[count++] = nums[i];
}
}
else
{
nums[count++] = nums[i];
}
}
while(count < len)
{
nums.pop_back();
count++;
}
return nums.size();
}
};
感觉代码太搓了,尤其是第一个if语句简直是一坨shit,和简洁优雅差太多。怎么样才能写出更优雅的代码。参考了下其他人的代码,发现其实只要将array中的array[i]与array[count-2]进行对比就可以了(i代表array遍历的索引,count代表在array上实时生成的新array的索引),如果相等则可以直接去除,否则就可以进行移位。这个方法写出了如此简洁的代码,简直就是TMD的艺术。
具体代码如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int count = 0;
int len = nums.size();
for(int i=0;i<len;i++)
{
if(count < 2 || nums[i] > nums[count-2])
nums[count++] = nums[i];
}
return count;
}
};