1.题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-element
2.解题思路
2.1 法一 暴力破解法
题目要求常数量级的空间复杂度,那就用空间换时间,暴力法。
第一层for循环寻找值等于val的元素nums[i],第二层for循环,从nums[i+1]寻找值不等于val的元素进行交换。
2.2 法二 双指针法
直接”重写“数组,right指针指向当前将要处理的元素,左指针 left 指向下一个将要赋值的位置。
如果rigth指向的元素不等于val,它是输出数组的一个元素,则重写数组,即将right指向的元素复制到left位置,同时,将左右指针同时右移;
如果右指针指向的元素等于val,需要将其跳过,不写入数组,此时left不动,right右移一位。
3.代码
//法一
class Solution {
public:
void swap(int &a1,int &a2) {
int tep=a1;
a1=a2;
a2=tep;
}
public:
int removeElement(vector<int>& nums, int val) {
int count=0;
int n=nums.size();
for(int t:nums) {
if(t!=val) {
count++;
}
}
for(int i=0; i<n; i++) {
if(nums[i]==val) {
for(int j=i+1; j<n; j++) {
if(nums[j]!=val) {
swap(nums[i],nums[j]);
}
}
}
}
return count;
}
};
//法二
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n = nums.size();
int left = 0;
for (int right = 0; right < n; right++) {
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
};