当前位置：首页 > article >正文

leetcode 数组 27.移除元素

article 2024/10/8 22:07:54

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：
更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。

不能直接删除，数组的元素在内存地址中是连续的，不能单独删除数组中的某个元素，只能覆盖

暴力解法，复杂度O（n2）：
两层for循环，一个for循环遍历数组元素，第二个for循环更新数组

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 + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};

下面我一开始的代码，不规范直接超时，即便是上面的也有超时不通过风险：

class Solution {
public:
int removeElement(vector& nums, int val) {
int count=0;
for(int i=0;i<nums.size();i++){
if(nums[i]==val){
for(int j=i;j<nums.size();j++){
nums[j]=nums[++j];
i–;
count++;
}
}
}return count;
}
};

双指针法（快慢指针法）：通过一个快指针和慢指针在一个for循环下完成两个for循环的工作

快指针：寻找新数组的元素，新数组就是不含有目标元素的数组
慢指针：指向更新新数组下标的位置

快慢指针，满指针便是新数组，快指针先走判断，直接把与val不同的覆盖前面的

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
            if (val != nums[fastIndex]) {
                nums[slowIndex++] = nums[fastIndex];
            }
        }
        return slowIndex;
    }
};