PTA:改写二分搜索算法

改写二分搜索算法

题目:
PTA:改写二分搜索算法_第1张图片

代码如下:


#include
using namespace std;
int binarySearch(int arr[], int x, int low, int high) {
    if (low > high) {
        return low;
    };
    int mid = (low + high) / 2;
    if (arr[mid] == x) {
        return mid;
    };
    if (arr[mid] > x) {
        return binarySearch(arr, x, low, mid - 1);
    }
    else {
        return binarySearch(arr, x, mid + 1, high);
    }
}

int main() {
    int len;
    int arr[100000];
    int x;
    cin >> len;
    cin >> x;
    for (int i = 0; i < len; i++) {
        cin >> arr[i];
    };
    int index = binarySearch(arr, x, 0, len - 1);
    if (arr[index] == x) {
        cout << index << " " << index << endl;
    }
    else {
        cout << index - 1 << " " << index << endl;
    }
    system("pause");
    return 0;
}

解析:
这道题我们需要输出x不在数组中时,返回小于x的最大元素位置i和大于x的最小元素位置j。只要对经典二分查找算法binarySearch函数进行改写,当元素不存在时,返回low即可。(图用left和right更明显)。
PTA:改写二分搜索算法_第2张图片
PTA:改写二分搜索算法_第3张图片
PTA:改写二分搜索算法_第4张图片
PTA:改写二分搜索算法_第5张图片
因为当low>high时即退出递归返回,所以此时low一定是刚好大于x的最小元素位置的下标(即j)。由题目可以看出,i和j的坐标一定是连续的。所以只要输出low和low-1即可。当元素存在时,返回mid下标并输出。

你可能感兴趣的:(算法)