题目与链接
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
添加链接描述
解题思路
这题目有两种解法,第一种是暴力嵌套遍历,外层遍历数组元素,依次找到符合条件的数组元素,内层对数组进行移位操作。这种解法的时间复杂度为O(n2)。第二种是采用双指针,在本题中和二分法有点相像,本质上都是二分类。二分法是以一个区间为单位,而双指针是以一个数组元素为单位。在这里,我们根据解题需要,将前向指针定义为寻找nums[ind]==val,将后向指针定义为寻找nums[ind]!=val。在双双觅得对象后,进行互换,接着进行下一次搜索。另外还是需要思考一下,边界条件的判断。我们思考一下,当start==end时会发什么,首先回顾两类指针的定义,前向指针是搜索==val的情况,也就是说在==val时,会停下,后向指针反之。而在最后会进行一次swap,并且前向指针++,后向指针-- 。这个时候如果前向指针指向的元素==val,那么直接退出,符合条件的数组元素数目正好为satrt,若是!=val,start++,然后退出,符合条件的数组元素数目依然为satrt。至此,本题的解题思路阐述完毕。
代码
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int start = 0, end = nums.size()-1;
while(start<=end){
while(start<=end && nums[start]!=val) {
++start;
}
while(start<=end && nums[end]==val) {
--end;
}
if(start<=end) nums[start++] = nums[end--];
}
return start;
}
};