题目链接:https://leetcode.cn/problems/remove-duplicates-from-sorted-array/solutions/
我一开始的解法很暴力,遇到和前一个元素相同的,就把该元素后面所有元素往前排。这样的做法很容易超时
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int pre = nums[0];
int size = nums.size();
for (int i = 1; i < size; i++) {
pre = nums[i-1];
if(nums[i] == pre) {
//delete
for(int j = i + 1; j < nums.size(); j++){
nums[j-1] = nums[j];
}
i--;
size--;
}
}
return size;
}
};
ac代码
双指针法,这个方法比较巧妙,一个指针p在前面,一个指针q在后面。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size() == 1) return 1;
int p = 0, q = 1;
while(q < nums.size()) {
if(nums[p] != nums[q]) {
nums[++p] = nums[q];
}
q++;
}
return p+1;
}
};