题目:27. 移除元素
解答:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int i=0,j=nums.size()-1;
int n=0;
if(nums.empty()==true)
return 0;
while(i<j){
if(nums[i]!=val){
i++;
}
else if(nums[i]==val){
while(j>=i&&nums[j]==val)
j--;
if(j<0||j<=i)
break;
int t=nums[i];
nums[i]=nums[j];
nums[j]=t;
i++;
}
}
for(int k=0;k<nums.size();k++){
if(nums[k]!=val)
n++;
}
return n;
}
};
比较愚蠢的双指针一头一尾遍历,过程中遇到了几次对于n的计数错误问题,最后索性再额外遍历了一遍。
以下为宫水三叶遍历一次的题解,非常简洁。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int idx = 0;
for(auto x : nums)
if(x != val)nums[idx++] = x;
return idx;
}
};作者:宫水三叶