27.移除元素

参考

https://programmercarl.com/0027.移除元素.html#思路

题目

---

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

---

1.暴力解法

---

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

点击查看代码

// 时间复杂度：O(n^2)
// 空间复杂度：O(1)
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;

    }
};

---

2.双指针

---

参考:https://leetcode.cn/problems/remove-element/solution/yi-chu-yuan-su-by-leetcode-solution-svxi/

---

点击查看代码

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
   int n=nums.size();
   //指向第一个元素的left的指针
   int left=0;
   for(int right=0;right<n;right++){
       if(nums[right]!=val){
           nums[left]=nums[right];
            left++;
       }
   }
   return left;
 }
};

---

3.双指针优化

---

点击查看代码

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;
    }
};

---

如果左指针 left 指向的元素等于 val，此时将右指针 right 指向的元素复制到左指针 left 的位置，然后右指针 right 左移一位。如果赋值过来的元素恰好也等于 val，可以继续把右指针 right 指向的元素的值赋值过来（左指针 left 指向的等于val 的元素的位置继续被覆盖），直到左指针指向的元素的值不等于val为止。

当左指针left和右指针right 重合的时候，左右指针遍历完数组中所有的元素。

这样的方法两个指针在最坏的情况下合起来只遍历了数组一次。与方法一不同的是，方法二避免了需要保留的元素的重复赋值操作。

链接：https://leetcode.cn/problems/remove-element/solution/yi-chu-yuan-su-by-leetcode-solution-svxi/