remove element

ayst123

于 2014-11-08 00:47:23 发布

阅读量363

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/ayst123/article/details/40909499

leetcode 专栏收录该内容

216 篇文章

订阅专栏

本文介绍了一种在原地(in-place)从数组中移除特定值的高效算法，并提供了C++及Python两种语言的实现代码。该算法通过一次遍历完成元素过滤，返回更新后的数组长度，忽略新长度之外的内容，适用于不关心多余位置数据的情况。

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

参考了网上

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.

原来数组可以in-place 替换，应该成为一个子问题

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int index = 0;
        for (int i=0;i<n;i++){
            if (A[i]!=elem){
                A[index++] = A[i];
            }
        }
        return index;
        
    }
};

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):
        index = 0
        for i in range(len(A)):
            if A[i]!=elem:
                A[index] = A[i]
                index += 1
        
        return index