题目
链接: 27.
题目描述:给你一个数组nums和一个值val,你需要原地移除所有数值等于val的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用O(1)额外空间并原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:为什么返回数值是整数,但输出的答案是数组呢?请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
示例 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,4,0,3]
解释:函数应该返回新的长度5, 并且nums中的前五个元素为0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
思路
这道题其实是比较简单的,其实和昨天的26题是一样的,也是使用双指针的方法进行。用一个指针来存储当前存储的位置,另一个指针存储当前检查到的元素位置。如果检查位置的数字不等于val,即将其存储到存储位置,并将存储位置的指针后移一位。这样可以达到O(N)的时间复杂度。
结果与不足之处
结果:执行用时:0ms,在所有 C++ 提交中击败了100%的用户;
内存消耗:8.6MB,在所有 C++ 提交中击败了25.66%的用户。
新知识
无
自己的代码
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int len = nums.size();
int i = 0, j = 0;
for(; i < len; i++){
if(nums[i] != val){
nums[j] = nums[i];
j++;
}
}
return j;
}
};
官方代码
因为可以改变顺序,所以有一种左右并进的双指针方式,代码如下。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left = 0, right = nums.size();
while (left < right) {
if (nums[left] == val) {
nums[left] = nums[right - 1];
right--;
} else {
left++;
}
}
return left;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/remove-element/solution/yi-chu-yuan-su-by-leetcode-solution-svxi/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。