给你一个 非严格递增排列 的数组 nums
,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
考虑 nums
的唯一元素的数量为 k
,你需要做以下事情确保你的题解可以被通过:
- 更改数组
nums
,使nums
的前k
个元素包含唯一元素,并按照它们最初在nums
中出现的顺序排列。nums
的其余元素与nums
的大小不重要。 - 返回
k
。
判题标准:
系统会用下面的代码来测试你的题解:
int[] nums = [...]; // 输入数组 int[] expectedNums = [...]; // 长度正确的期望答案 int k = removeDuplicates(nums); // 调用 assert k == expectedNums.length; for (int i = 0; i < k; i++) { assert nums[i] == expectedNums[i]; }
如果所有断言都通过,那么您的题解将被 通过。
示例 1:
输入:nums = [1,1,2] 输出:2, nums = [1,2,_] 解释:函数应该返回新的长度2
,并且原数组 nums 的前两个元素被修改为1
,2
。
不需要考虑数组中超出新长度后面的元素。
示例 2:
输入:nums = [0,0,1,1,1,2,2,3,3,4] 输出:5, nums = [0,1,2,3,4] 解释:函数应该返回新的长度5
, 并且原数组 nums 的前五个元素被修改为0
,1
,2
,3
,4
。不需要考虑数组中超出新长度后面的元素。
提示:
1 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums
已按 非严格递增 排列
我的答案如下:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int len = nums.size();
vector<int>b;
vector<int>::iterator it = nums.begin();
for (int i = 0; i < len-1; i++) {
if (*(it + i) ==*(it + i + 1))
b.push_back(i);
}
for(int i=b.size()-1;i>=0;i--)
{
nums.erase(nums.begin()+b[i]);
}
for(int i=0;i<nums.size();i++)
cout<<nums[i]<<" ";
return nums.size(); // 添加返回值
}
};
答案如下:(使用algorithm库不算英雄)。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
return distance(nums.begin(),unique(nums.begin(),nums.end()));
}
};
这里面用到两个新的函数:
unique函数:同remove,将重复元素移动到向量末尾,然后返回一个指向新逻辑结尾的迭代器,此时,nums前半部分中的元素是没有相同元素的。
distance用于返回两个参数之间的距离,也就是从nums.begin()开始到nums.end()之间的距离,即元素个数。