给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。
用户评测:评测机将使用以下代码测试您的解决方案:
int[] nums = [...]; // 输入数组
int val = ...; // 要移除的值
int[] expectedNums = [...]; // 长度正确的预期答案。
// 它以不等于 val 的值排序。int k = removeElement(nums, val); // 调用你的实现
assert k == expectedNums.length;
sort(nums, 0, k); // 排序 nums 的前 k 个元素
for (int i = 0; i < actualLength; i++) {
assert nums[i] == expectedNums[i];
}
如果所有的断言都通过,你的解决方案将会 通过。示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2,_,_]
解释:你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
示例 2:输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,4,0,3,_,_,_]
解释:你的函数应该返回 k = 5,并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要(因此它们并不计入评测)。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
对需求进行分析,因为不需要对右侧的元素进行考虑,所以交换的时候直接左侧等于右侧就行,先设置一个左指针,再设置一个右指针等于数组的最右侧也就是nums.size()-1(注意当数组为空的时候就是-1)<当且仅当左侧指针大于右侧指针的时候退出循环,因为要求的是元素个数。
进入循环,当左指针指向的元素不等于val那么就对左指针进行++操作向后移动,表明这些元素都是需要保存的,当等于val的时候就需要和右边的进行交换,此时右边的可能也为val,所以交换之后只对右指针进行--操作,对左指针再进行一次判断,看其是否等于val,如果等于val,再继续进行判断操作。
主要是对边界的考虑,当左侧指针刚好等于右侧指针的时候,此时判断是否还等于val,如果等于val就不需要再对做操作,右指针向左移动,左指针不动,左指针之前的元素都是不等于val的元素(不包含左指针指向的元素)。如果不等于val的话,左指针向后移动,左指针之前的元素都是不等于val的元素(不包含左指针指向的元素)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left = 0;
int right = nums.size()-1;
while(left<=right){
if(nums[left] != val){
left++;
}
else{
nums[left] = nums[right];
right--;
}
}
return left;
}
};