算法02数组-移除元素

题目

给你一个数组 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。

你不需要考虑数组中超出新长度后面的元素。

思路

01 暴力

发现需要的值，就让后面的每一位向前覆盖

02 双指针

原地修改数组，相当于用自己覆盖自己

相较于01，双指针算法可能不好理解，笔者用一个视频演示第一次做这种视频，见谅

111

代码实现

01

class Solution {
public:
    int removeElement(vector<int>& nums, int val){
        int i,j,size=nums.size();
        for(i=0;i<size;i++){
            if(nums[i]==val){//如果碰到目标值就从该值起，让后面一个覆盖前面一个，注意j的边界
                for(j=i;j<size-1;j++){
                    nums[j]=nums[j+1];
                }
                size--;//size动态减少
                i--;//i对应的值发生改变，重新对该位置判断
            }
        }
    return size;//返回长度
    }
};

02

//双指针
class Solution{
public:
      int removeElement(vector<int>& nums, int val){
          int fast,slow=0;//设置快慢指针，慢指针初始化为0
          for(fast=0;fast<nums.size();fast++){
              if(nums[fast]!=val){
                  nums[slow]=nums[fast];//一旦发现不是目标值
                  slow++;//就存入慢指针所对应的下表，slow对应更新
              }
          }
      return slow;//slow是慢指针下标的下一位，也就是数组的长度
      }
};

思考

本题采用两种不同的解法，在力扣上都可以通过，不过使用双指针减少了一层循环，降低了时间复杂度由O(n*n)变为O(n)。为双指针的进阶打下了基础，希望各位友友们多加练习，跟笔者共同进步哦~

尾声

前面的二分搜索算法结束啦，希望大家勤联系，多加思考，掌握算法思维，后面会更新数组相关的算法知识，如果觉得笔者讲的不错的话，希望大家点赞，关注，收藏支持以下噢~