LeetCode - Remove Element

最新推荐文章于 2024-03-25 22:40:48 发布

原创最新推荐文章于 2024-03-25 22:40:48 发布 · 253 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#array #two pointers

leetcode 专栏收录该内容

127 篇文章

订阅专栏

本文介绍了一种在数组中移除指定元素并返回新长度的高效算法。通过一次遍历和使用指针记录非目标元素的方法，实现了O(n)的时间复杂度和O(1)的空间复杂度。

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

https://leetcode.com/problems/remove-element/

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.

这道题就是遍历一遍，维持一个指针，指针前面的位置都不是elem，如果碰到不是elem的值，就把这个值复制到指针所在位置，指针前移，如果碰到elem，就不复制，继续遍历数组。代码如下：

    public int removeElement(int[] A, int elem) {
        if(A==null || A.length==0) return 0;
        int current = 0;
        for(int i=0; i<A.length; i++){
            if(A[i]==elem) continue;
            else{
                A[current] = A[i];
                current++;
            }
        }
        return current;
    }

时间复杂度O(n)，空间复杂度O(1)