题目链接:
https://leetcode.cn/problems/remove-duplicates-from-sorted-array/description/?envType=study-plan-v2&envId=top-interview-150
题目描述:
解题思路:
使用双指针算法(快慢指针),p1与p2分别指向第一个和第二个元素,p2为快指针,p1为慢指针。
- 当
nums.size() == 0
时,直接return 0;
即可 - 当
nums.size() !=0
时,p1作为慢指针指向第一个元素,p2作为快指针指向第二个元素,res 初始化为1 - 当
nums[p1] == nums[p2]
时,只需要将p2++
即可 - 当
nums[p1] != nums[p2]
时,只需要将p+1
位置的元素与p2交换,同时指针p1和p2都移向下一个元素,由于此时优找到了一个新元素,因此res也要加1 - 当p2指针超过
nums.size()
,说明数组遍历完成,结束循环
代码实现:
时间复杂度O(n)
空间复杂度O(1)
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int n = nums.size();
if (n==0){
return 0;
}
int p1 = 0;
int p2 = 1;
int res = 1;
while(p2<n){
if(nums[p1] == nums[p2]){
p2++;
}
else{
p1++;
nums[p1] = nums[p2];
p2++;
res++;
}
}
return res;
}
};