leetcode 27 Remove Element

删除数组中特定元素并返回新长度的高效算法

最新推荐文章于 2025-09-07 21:00:40 发布

weixin_34005042

最新推荐文章于 2025-09-07 21:00:40 发布

阅读量58

点赞数

CC 4.0 BY-SA版权

文章标签： c/c++ python

原文链接：https://yq.aliyun.com/articles/257233

本文介绍了一种删除数组中特定元素并返回新长度的方法，包括C++、Python和最简洁的解决方案。算法通过迭代遍历数组，将非目标元素移动到数组的前部，并在循环结束后返回新数组的有效长度。

Remove Element Total Accepted: 60351 Total Submissions: 187833 My Submissions

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

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

Show Tags

c++ 解决方案：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        int i = 0;
        while( i < n ) { 
            if( nums[i] == val ) {
                swap(nums[i], nums[n-1]);
                n--;
            } else {
                i++;
            }
        }
        return n;
    }
};

int removeElement(vector<int>& nums, int val)
{
    vector<int>::iterator  itr = nums.begin();
    while (itr != nums.end())
    {
        if (*itr == val)
            itr = nums.erase(itr);
        else
            ++itr;
    }
    return nums.size();
}

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

python解决方案：

class Solution:
# @param    A       a list of integers
# @param    elem    an integer, value need to be removed
# @return an integer
def removeElement(self, A, elem):
    i = 0
    for j in range(len(A)):
        if A[j] != elem:
            A[i] = A[j]
            i += 1
    return i

史上最简洁的解决方案：

def removeElement(self, nums, val):
        nums[:] = [x for x in nums if x!=val]
        return len(nums)