使用方法有暴力法和双指针法,关键点在于快慢指针的含义和作用。
暴力法:时间复杂度:O(log n^2),空间复杂度:O(1)
暴力法思路:
- 构建for循环,第一层循环遍历数组,查找相同的元素。
- 在查找到相同元素后,构建第二层for循环,将相同元素坐标后的所有元素向前移动一位,同时减少size大小。注意,在将所有元素向前移动一位后,nums[i]变成了原本的nums[i+1],所以需要将i--,以重新检查现在的nums[i]是否与val相等。
暴力法代码:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (nums[i] == val) {
for (int j = i; j < size - 1; j++) {
nums[j] = nums[j + 1];
}
size--;
i--; //重新检查下标i的新数值,因为下标i以后的数值都向前移动了一位
}
}
return size;
}
};
双指针法:时间复杂度:O(log n),空间复杂度:O(1)
双指针法思路:
- 构建快慢指针,初始值都指向数组的开头。快指针用来帮助寻找需要保留的值,慢指针用来帮助保存需要保留的值,同时慢指针本身是保存下来的数组元素的个数。
- 构建一层for循环,比较快指针指向的值是否与目标值相等。如果不相等(说明这个值是需要保留的值),将快指针指向的位置的值赋给慢指针指向的位置,然后慢指针指向下一个位置。如果相等(说明这个值是需要删除的),慢指针不动,快指针正常进入下一个循环。
- 因为慢指针在赋值后++了,所以慢指针是指向的保存下来的新数组的后面一个位置,刚好就是保存下来的数组元素的个数(数组位置是从0开始的,个数是从1开始的)。所以,直接返回慢指针。
双指针法代码:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int fastIndex = 0;
int slowIndex = 0;
for (fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
if (nums[fastIndex] != val) {
nums[slowIndex] = nums[fastIndex];
slowIndex++;
}
}
return slowIndex;
}
};