0
点赞
收藏
分享

微信扫一扫

【LeetCode27】移除元素(双指针)


1.题目

【LeetCode27】移除元素(双指针)_算法

2.思路

要原地去掉数值为val的数,最浅显的暴力做法就是遍历到该数时直接依次将数组后面的数字往前移动一个,如下:

class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (nums[i] == val) { // 发现需要移除的元素,就将数组集体向前移动一位
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下表i以后的数值都向前移动了一位,所以i也向前移动一位
size--; // 此时数组的大小-1
}
}
return size;

}
};

但是力扣上会超时,链表和数组题我们经常使用双指针,这题可以设置快慢指针——只for遍历一次,快指针i冲在前,而如果遇到目标val值时,慢指针j就歇一格,并且将快指针i所在的元素复制到慢指针j所在的位置上;

而当没有遇到目标val值就i和j同并进。

【LeetCode27】移除元素(双指针)_算法_02

3.代码

class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size=nums.size();
int j=0;//慢指针
for(int i=0;i<size;i++){//i为快指针
if(nums[i]!=val){
nums[j++]=nums[i];
}
}
return j;
}
};


举报

相关推荐

0 条评论