源自力扣题解
一、题目描述
给一个数组 nums 和一个值 val,原地 移除所有数值等于 val 的元素,并返回移除后数组新长度。
注意:
- 不用额外的数组空间,仅使用 O(1) 额外空间并 原地 修改输入数组。
- 元素的顺序可以改变。不需要考虑数组中超出新长度后面的元素。
力扣题目编号 27,属于简单类
二、示例
示例 1
输入:nums = [3,2,2,3], val = 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], val = 2
输出:5, nums = [0,1,3,0,4]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
三、思路
题目要求不考虑数组超出新长度后面的元素,也就是说,把不等于 val 的全覆盖到前面去就行了
四、题解
解法 1 —— 这个最快
看到一个很妙的解法,除了很妙的思路外,还要记住自增的妙用
int removeElement(int* nums, int numsSize, int val) {
int newSize = 0; // 新数组长度
for (int i = 0; i < numsSize; i++)
{
if (nums[i] != val) {
//若不等于 val ,则从下标0起原地覆盖数组,至所有元素检索完为止
nums[newSize++] = nums[i];
}
}
return newSize;
}
解法 2
两个解法从代码上看是一样的,只不过思路上用了双指针,这也是官方给的解答
左右两个指针,左指针指向更新位,右指针遍历寻找新数组元素 —— 不等于 val 的元素
int removeElement(int* nums, int numsSize, int val) {
int left = 0;
for (int right = 0; right < numsSize; right++) {
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
解法 3
设置左右首尾指针,若左指针所指元素等于 val ,则,将右指针所指元素,赋给左指针所指之处,而后右指针前移;若左指针所指元素不为 val ,则,左指针后移。直至左、右指针相遇为止。
如此一来,从下标 0 起至 左指针-1,全都是不等于 val 的值,即新数组元素;而 left 到 right 之间是未处理过的元素。
左边出来一个 val,就把右边尾巴赋过去,尾指针前移;判断左边赋过去的值是否 = val,不等再指针后移;相等则再把尾巴赋过去,跟前面一样。所以 left + 1 = right 时,0 至 left-1
是所求新数组元素,left 为 新数组长度
int removeElement2(int* nums, int numsSize, int val) {
int left = 0, right = numsSize - 1;
while (left <= right) {
if (nums[left] == val)
nums[left] = nums[right--];
else
left++;
}
return left;
}
只是这个解法,不保证稳定性,即可能破坏元素相对顺序。