程式碼隨想錄2

continu~發表於2024-10-14

題目:移除元素:
給你一個陣列 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;
}

相關文章