我想从排序数组中返回某个元素的位置(如果该元素存在),否则我想返回应该插入该元素的插入位置。 这是我的代码:
public static int bs(int[] array, int left, int right, int elem) {
if (left > right) {
return left;
} else {
int middle;
middle = (left + right) / 2;
if (left == right) { // used to return the insert position
return left;
} else if (elem > array[middle]) {
return bs(array, middle + 1, right, elem);
} else if ((elem < array[middle])) {
return bs(array, left, middle - 1, elem);
} else {
return middle; // element existed into array
}
}
}
例如:
array = [ 2 5 8 10], elem = 8
=>将返回2
array = [ 2 5 8 10], elem = 6
=>将返回1
array = [ 2 7 14 22 32 56 88 91 102], elem = 3
=>将返回1(但以上程序返回0)
答案 0 :(得分:1)
您的错误是从bs(left,middle-1)的拆分中删除array [middle]。相反,您需要使用bs(left,middle)
和bs(middle+1,right)
。我将打印添加到递归调用中,并轻松找到它。
public static int bs(int[] array, int left, int right, int elem) {
System.out.println("["+left+", "+right+"]");
if (left > right) {
return left;
} else {
int middle;
middle = (left + right) / 2;
if (left == right) { // used to return the insert position
return left;
} else if (elem > array[middle]) {
return bs(array, middle + 1, right, elem);
} else if ((elem < array[middle])) {
return bs(array, left, middle, elem); //<-- was: middle-1
} else {
return middle; // element existed into array
}
}
}
我也认为这种风格会更好;)
public static int bs2(int[] array, int left, int right, int elem) {
System.out.println("["+left+", "+right+"]");
if (left >= right)
return left;
int middle;
middle = (left + right) / 2;
if (elem > array[middle])
return bs(array, middle + 1, right, elem);
if ((elem < array[middle]))
return bs(array, left, middle, elem); //<--- was: middle-1
return middle; // element existed into array
}
答案 1 :(得分:0)
我希望我对你没错。 这是我的第一篇文章,所以请怜悯我:)。
起初我认为 数组= [2 5 8 10],elem = 6 =>将返回2是正确的, 因为6> 5,所以它的索引为2。 还是要用6代替5? 如果要替换,请删除我的评论的“ //”。
public static int getPosition(int[] arr, int value)
{
int position = 0;
for (int i = 0; i < arr.length; i++)
{
if (arr[i] == value)
{
position = i;
break;
} else
position = -1;
}
if (position == -1)
{
int[] arr2 = new int[arr.length + 1];
System.arraycopy(arr, 0, arr2, 0, arr.length);
arr2[arr.length] = value;
Arrays.sort(arr2);
position = getPosition(arr2, value);//-1;
}
return position;
}
public static void main(String[] args)
{
int[] arr = { 2, 5, 8, 10 };
System.out.println(getPosition(arr, 6));
}