题目描述
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
例子
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
说明
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
分析
- 有序数组,特点是重复元素都挨在一起
- 比较元素是否和上一个重复,如果重复就删除一个元素
- 要求O(1)的空间,所以可以用时间换空间
- 原地删除修改,我的理解是:用原有的数组,不用辅助数组
- 不需要考虑数组中超出新长度后面的元素,也就是说后面的元素不用修改,只用直接赋值到前面即可
算法思路
- 设置一个变量prev(previous),记录当前数组中元素;另一个变量记录不同的个数,也就是数组新的长度len(length)
- 比较下一个元素是否相同,直到不相同,将这个不相同的元素,重新赋值到第len的位置,即更改了此位置的元素数值,len++
- 接着将当前比较元素prev更新为此不相同的元素,继续向后比较下一个元素
- 返回新的长度
代码
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int removeDuplicates(vector<int>& nums)
{
//特殊情况:空数组
if (nums.size() == 0)return 0;
int prev=nums[0];
int len=1;
for(int k=1 ;k < nums.size();k++)
{
//遇到不同元素
if( nums[k] != prev)
{
nums[len]= nums[k];
len++;//不同元素的个数增加
prev = nums[k];//prev向后比较下一个元素
}
}
return len;
}
};
int main() {
Solution A;
vector<int> str={0,0,1,1,1,2,2,3,3,4};
cout << A.removeDuplicates(str) << endl;
return 0 ;
}
总结
- vector
& nums的使用 - 善于用条件语句,遇见!=的条件比遇见==的条件,情况更少
- 注意顺序:特殊情况一定在最前面,否则会出错:因为空数组没有nums[0],但是代码中用来赋值了;
结果
执行用时 :16 ms, 在所有 C++ 提交中击败了81.18%的用户
内存消耗 :7.7 MB, 在所有 C++ 提交中击败了100.00%的用户