LeetCode 027 Remove Element

最新推荐文章于 2025-09-06 17:22:57 发布

原创最新推荐文章于 2025-09-06 17:22:57 发布 · 779 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #面试笔试

LeetCode 专栏收录该内容

138 篇文章

订阅专栏

本文介绍了一种高效的方法来去除数组中的特定值，并在原地修改数组的同时返回新的有效长度。通过双指针技术实现，分别从头部和尾部进行元素检查与交换，最终达到快速过滤指定值的目的。

题目

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.

去掉一个数组中的某个值，返回新的数组长度。

思路

1 题目要求是in place

2 直接想到了头尾分割的方法。把所有碰到的元素都放到尾巴，然后获得新的length即可。

代码

public class Solution {
    public int removeElement(int[] A, int elem) {
        if(A.length==0){
            return 0;
        }
        
        int cur =0;
        int end = A.length-1;
        
        while(cur<=end){
            if(A[cur]!=elem){
                cur++;
            }
            else{
                int temp =A[cur];
                A[cur]=A[end];
                A[end]=temp;
                end--;
            }
        }
        return cur;
    }
}