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