[LeetCode] Remove Element题解

最新推荐文章于 2025-05-03 18:08:53 发布

原创最新推荐文章于 2025-05-03 18:08:53 发布 · 193 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一个常见的算法问题——RemoveElement，即如何在不使用额外数组的情况下移除数组中指定值的所有实例，并返回新长度。文章提供了两种解决方案：一种利用C++ STL中的erase函数，另一种通过迭代和覆盖的方式实现。这两种方法均遵循题目要求，使用常数级额外空间完成操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Remove Element：

Given an array and a value, remove all instances of that value in place and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

The order of elements can be changed. It doesn’t matter what you leave beyond the new length.

Example:
Given input array nums = [3,2,2,3], val = 3

Your function should return length = 2, with the first two elements of nums being 2.

这道题如果没有其他限制，另开一条数组，将不等于val的数存进去就ok了，但是它限制了我们另开一条数组，所以可以用c++ STL中vector的erase函数直接在原数组中更改：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        while(remove(nums,val)){
        }       
        return nums.size();
    }
    bool remove(vector<int>& nums, int val){
        vector<int>::iterator iter;
        for(iter=nums.begin();iter!=nums.end();iter++){
            if(*iter==val){
                nums.erase(iter);
                return true;
            }
        }
        return false;
    }
};

而在一些语言中没有erase函数可以用，这么办？不用erase函数的话，我们可以这样处理：

int removeElement(vector<int>& nums, int val) {
    int cnt = 0;
    for(int i = 0 ; i < nums.size() ; ++i) {
        if(nums[i] == val)
            cnt++;
        else
            nums[i-cnt] = nums[i];
    }
    return nums.size()-cnt;
}

或者

int removeElement(int A[], int n, int elem) {
    int begin=0;
    for(int i=0;i<n;i++){
        if(A[i]!=elem){
            A[begin++]=A[i];
        }
    }
    return begin;
}