给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {//传入的是引用
if(nums.size()<2){//一个小坑,没有对特殊案例进行思考
return nums.size();
}
sort(nums.begin(),nums.end());//先排序
int pos = 0;
int i ;
for( i =0;i<nums.size()-1;i++){
if(nums[i]!=nums[i+1]){
nums[++pos] = nums[i+1];
}
}
return pos+1;
}
};
如果没有一开始的那个if,那么如果给出的示例是 [] 时 ,即,size0,显然会出错。那么事先排除size0 、 size == 1的情况即可。
显然保留第一个元素(不可能重复嘛,故 i 从1 开始即可),然后从第二个位置逐个与前面比较,不相同就保留。保留的位置是pos。
最后输出数组长度 size (即pos+1) ;