【题目描述】
给你一个数组 nums
和一个值 val
,你需要原地移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
【测试用例】
示例1:
输入:nums = [3,2,2,3]
输出:2,nums = [2,2]
解释:函数应该返回新的长度2,并且 nums 中的前两个元素均为2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为2,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例2:
输入:nums = [0,1,2,2,3,0,4,2]
输出:5,nums = [0,1,3,0,4]
【思路分析】
这道题和LeetCode简单题的26题异曲同工,不同之处在于循环遍历数组nums时候内部的判断条件不同和pos1初始位置、后移条件不同。具体的可以参考我的26题题解并结合本题代码。
【参考代码】
C实现
#include <stdio.h>
//easy-27-移除元素
int removeElement(int* nums, int numsSize, int val);
int main(){
int nums[] = {3,2,2,3};
int numsSize = sizeof(nums)/sizeof(nums[0]);
int val = 3;
int k = removeElement(nums, numsSize, val);
int i;
printf("%d\n", k);
for(i=0;i<k;++i){
printf("%d ",nums[i]);
}
return 0;
}
int removeElement(int* nums, int numsSize, int val) {
int pos1=0, pos2=0;
while(pos2<numsSize){
if(nums[pos2] == val){
pos2++;
}else{
nums[pos1++] = nums[pos2++];
}
}
return pos1;
}
C++实现
#include <iostream>
#include <vector>
using namespace std;
//easy-27-移除元素
class Solution {
public:
int removeElement(vector<int>& nums, int val);
};
int Solution::removeElement(vector<int>& nums, int val){
int pos1=0, pos2=0;
while(pos2<nums.size()){
if(nums[pos2] == val){
pos2++;
}else{
nums[pos1++] = nums[pos2++];
}
}
return pos1;
}
int main(){
vector<int> nums = {3,2,2,3};
int val = 3;
Solution sol;
int k = sol.removeElement(nums, val);
int i;
cout<<k<<endl;
for(i=0;i<k;++i){
cout<<nums[i]<<" ";
}
return 0;
}