这是我真正开始刷题的第一步,也是第一篇博客,还是第一次遇到用双指针解决问题。此后将用博客来进行刷题记录,并且记录我的心得体会与学习经验。
我的代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<int> a;
if(nums.size() != 0)a.push_back(nums[0]);
for(int i = 1; i < nums.size(); i ++ ){
if(nums[i] == a.back())continue ;
else a.push_back(nums[i]);
}
int len = a.size();
for(int i = 0; i < len; i ++ ){
nums[i] = a[i];
}
return len;
}
};
我的第一反应是用vector来解决问题,首先将数组排序,将第一个元素插入数组,然后遍历,只要原数组的元素和a数组最后一个元素不同,就插入,否则就跳过,判断下一个数。思路简单,没有算法。
接着分析官方给的思路
为什么要用双指针呢?因为题目描述中有关键字“原地修改”,也就是说,对于两个重复的元素,只需要覆盖掉后一个元素,保留前一个,就可以对原数组进行“原地修改”。
如何实现?
这就要考虑到怎么用双指针了,此题中,定义一个快指针,一个慢指针,快指针用来扫描数组,慢指针用来覆盖元素。开始时两指针都指向第一个元素,当快指针扫描到与之前的元素不同的元素时,慢指针向前移动一位,并且把新元素进行覆盖,最终完成题目。另外,不要忘记考虑数组长度为0的情况!!!
对于复杂度上,我的代码和答案都要对数组进行一次扫描,因此时间复杂度都为O(n),但是我的代码空间复杂度为O(n),而答案的空间复杂度为O(1)
双指针的代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size() == 0)return 0;
int k = 0;//慢指针
for(int i = 0; i < nums.size(); i ++ ){//i为快指针
if(nums[i] != nums[k]){
k++;
nums[k] = nums[i];
}
}
return (k + 1);
}
};