LeetCode Remove Element 解题报告

最新推荐文章于 2024-06-25 21:29:50 发布

原创最新推荐文章于 2024-06-25 21:29:50 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #面试题 #算法 #Java

面试题同时被 3 个专栏收录

51 篇文章

订阅专栏

Java

44 篇文章

订阅专栏

算法

43 篇文章

订阅专栏

本文介绍了两种删除数组中等于特定值的元素，并返回新数组大小的方法。第一种方法通过前后指针交换元素，考虑了各种边界条件；第二种方法采用双指针策略，将不等于特定值的元素移动到新位置，优化了内存使用。

http://oj.leetcode.com/problems/remove-element/

删除数组中等于某value的元素，元素有可能有多个，返回新数组的大小。

基本思路：从前向后找到等于value的元素，从后向前找到第一个不等于value的元素，交换。

AC代码：

public static int removeElement(int[] A, int elem) {

        int i = 0;
        int temp;
        int end = A.length;
        for(i=0;i<A.length&&i<end;i++) {
            if(A[i]!=elem) {
                continue;
            }
            end--;
            while(A[end]==elem&&end>i) {
                end--;
            }
            if (end==i) {
                return i;
            }
            temp   = A[i];
            A[i]   = A[end];
            A[end] = temp;
        }
        return i;
}

这种方法对于各种边界条件要考虑的比较清楚才行，比较容易写错，我也是Wrong Answer好多次之后才AC。
还有一种方法是双指针，一个是在前遍历用的i，一个是在后待写入的start。i会越来越快，start慢一点。把不等于elem的元素都移动到start指向的位置。

public int removeElement(int[] A, int elem) {
	int start = 0;
	for(int i=0;i<A.length;i++) {
		if(A[i]==elem) {
			A[start++] = A[i];
		}
	}
	return start;
}