代码随想录2

jiyuebo / 2024-10-15 / 原文

题目：移除元素：
给你一个数组 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。

采用双指针的方式。将值等于val的元素移动到数组后面。
右指针记录当前可供后移的位置。左指针记录当前查看的元素。如果当前元素==val，则置换到右指针所指位置。右指针前移指向下一个可供替换的位置。
否则左指针后移，查看下一个元素。

#include<iostream>
#include<vector>
using namespace std;


int removeElement(vector<int>&nums,int val)
{

    int right=nums.size()-1;
    int current=0;

    while(current<=right)
    {
        if (nums[current]==val)
        {
            int temp=nums[right];
            nums[right]=nums[current];
            nums[current]=temp;
            right--;
        }
        else
        {
            current++;
        }
    }
    return right+1;
}

int main()
{

    vector<int>nums={0,1,2,2,3,0,4,2};
    int result=removeElement(nums,2);
    cout<<result<<endl;
    for(int i =0;i<nums.size();i++)
    {
        cout<<nums[i]<<endl;
    }
    return 0;
}