leetcode 27 : Remove Element
题目链接: https://leetcode-cn.com/problems/remove-element/
难度: 简单
归类 : 数组操作
题目:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例:
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
解法:
个人解法
首先判断空数组,随后根据指针一步步的往后迭代,遇到val就使用erase()函数将其删去,然后将指针-1(因为删去了一个,所以指针需要往后退一位),最后返回数组大小即可。
C++解法
#c++解法
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int len = nums.size();
if(len == 0 || (len == 1 && nums[0] == val)){
return 0;
}
vector<int>::iterator it;
for(it = nums.begin(); it != nums.end(); it++){
if(*it == val){
nums.erase(it);
it--;
}
}
return nums.size();
}
};
时间复杂度: O(N)
空间复杂度: O(1)
提交结果:
执行用时:0 ms, 在所有 C++ 提交中击败了100%的用户
内存消耗:6.5 MB, 在所有 C++ 提交中击败了9.55%的用户
题解优解
使用双指针法,可将暴力解决法的O(N^2)复杂度变为O(N)。
#c++解法
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slowIndex = 0; // index为 慢指针
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) { // i 为快指针
if (val != nums[fastIndex]) { //将快指针对应的数值赋值给慢指针对应的数值
nums[slowIndex++] = nums[fastIndex]; //注意这里是slowIndex++ 而不是slowIndex--
}
}
return slowIndex;
}
};
时间复杂度: O(N)
空间复杂度: O(1)
提交结果:
执行用时:4 ms, 在所有 C++ 提交中击败了55.37%的用户
内存消耗:6.3 MB, 在所有 C++ 提交中击败了72.52%的用户
尝试改进的最优解法
本题较为简单,无更优解法