使用Binary Search Recursion将元素的插入位置返回到排序数组中

时间:2018-06-28 07:34:30

标签: java binary-search

我想从排序数组中返回某个元素的位置(如果该元素存在),否则我想返回应该插入该元素的插入位置。 这是我的代码:

    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)

2 个答案:

答案 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));

}