題目:移除元素:
給你一個陣列 nums 和一個值 val,你需要 原地 移除所有數值等於 val 的元素,並返回移除後陣列的新長度。
不要使用額外的陣列空間,你必須僅使用 O(1) 額外空間並原地修改輸入陣列。
元素的順序可以改變。你不需要考慮陣列中超出新長度後面的元素。
示例 1: 給定 nums = [3,2,2,3], val = 3, 函式應該返回新的長度 2, 並且 nums 中的前兩個元素均為 2。 你不需要考慮陣列中超出新長度後面的元素。
示例 2: 給定 nums = [0,1,2,2,3,0,4,2], val = 2, 函式應該返回新的長度 5, 並且 nums 中的前五個元素為 0, 1, 3, 0, 4。
採用雙指標的方式。將值等於val的元素移動到陣列後面。
右指標記錄當前可供後移的位置。左指標記錄當前檢視的元素。如果當前元素==val,則置換到右指標所指位置。右指標前移指向下一個可供替換的位置。
否則左指標後移,檢視下一個元素。
#include<iostream>
#include<vector>
using namespace std;
int removeElement(vector<int>&nums,int val)
{
int right=nums.size()-1;
int current=0;
while(current<=right)
{
if (nums[current]==val)
{
int temp=nums[right];
nums[right]=nums[current];
nums[current]=temp;
right--;
}
else
{
current++;
}
}
return right+1;
}
int main()
{
vector<int>nums={0,1,2,2,3,0,4,2};
int result=removeElement(nums,2);
cout<<result<<endl;
for(int i =0;i<nums.size();i++)
{
cout<<nums[i]<<endl;
}
return 0;
}