题目:已知一个数组A,和一个值elem,需要把数组中的elem值全部删除,并且返回删除后的数组长度。为删除前数组的顺序可以任意改变的。
思路:先把数组进行排序,然后用2分查找的方法找到,需要删除值的位置,因为是2分法,所以它的前后都有可能有重复的,需要在往前后便利,找到相同的值后,用后面的数组覆盖这些重复的值就行了。
public int removeElement(int[] A, int elem) {
// IMPORTANT: Please reset any member data you declared, as
// the same Solution instance will be reused for each test case.
Arrays.sort(A);// 必须先排序要不出现{0,4,4,0,4,4,4,0,2}类型的a,得不到正确的答案了。
int start = 0;
int end = A.length - 1;
int location = 0;
while (true) {
if (start > end) {
return A.length;
}
location = (start + end) / 2;
if (A[location] == elem)
break;
if (A[location] > elem)
end = location - 1;
if (A[location] < elem)
start = location + 1;
}
int left = 0;// 察看左边有几个重复的
int index = location;
while (index > 0 && A[index] == A[index - 1]) {
left++;
index--;
}
index = location; // 察看右边有多少重复的
int right = 0;
while (index < A.length - 1 && A[index] == A[index + 1]) {
right++;
index++;
}
int length = right + left + 1;// 重复的长度。左边加上右边,还需要加上1,
// 这其实就是小明前面1个人,后面2个人,一共是4个人的道理
int begin = location - left;
for (int i = begin; i < A.length - length; i++) {
A[i] = A[i + length];
}
return A.length - length;
}