LeetCode Remove Element 解题报告

来源:互联网 发布:数据的物理模型 编辑:程序博客网 时间:2024/05/10 00:15

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


0 0
原创粉丝点击