27--移除元素
最开始想到的是单指针,但是无法处理连续相同的目标值,故想到双指针法,设立首尾指针,先判断数组是否为空,其次判断尾指针所指元素是否为目标值,是则使用vector的内置函数去除尾元素,并让尾指针重新指向末尾元素;否则判断首指针的元素是否为目标值,若不是,使首指针后移;若是,交换首尾指针所指元素,去除尾元素,重新赋值尾指针。在跳出循环后,判断数组大小,若数组只有一个,且元素为目标值,可能原数组所有元素都相等,直接返回0,反之返回数组大小。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int start = 0, end = nums.size() - 1;
while(start < end){
if(nums.empty()){
break;
}
if(nums[end] == val){
nums.pop_back();
end = nums.size() - 1;
continue;
}
else{
if(nums[start] == val){
swap(nums[start], nums[end]);
nums.pop_back();
end = nums.size() - 1;
}
else{
start++;
}
}
}
if(start == end && nums[start] == val){
return 0;
}
return nums.size();
}
};