计蒜客 寻找插入位置 （二分查找）

给定一个已经升序排好序的数组，以及一个数 target，如果 target 在数组中，返回它在数组中的位置。

否则，返回 target插入数组后它应该在的位置。

假设数组中没有重复的数。以下是简单的示例：

[1,3,5,6], 5 → 2

[1,3,5,6], 2 → 1

[1,3,5,6], 7 → 4

[1,3,5,6], 0 → 0

输入格式

第一行输入一个整数 n。

第二行输入 n 个整数，表示数组A[n]。

第三行输入 target。

输出格式

输出一行，为要求返回的结果。

样例输入

3
1 3 5
2

样例输出

1

#include <iostream>
using namespace std;

int binarySearch(int *a, int left, int right, int target){
    int middle;
    //target在数组中
    while(left <= right){
        middle = (right - left) / 2 + left;
        if(target == a[middle])
            return middle;
        else if(target < a[middle])
            right = middle - 1;
        else 
            left = middle + 1;
    }
    //target不在数组中，而且此时left == right
    if(target < a[middle])
        return middle;
    else
        return middle + 1;
}
int main(){
    int n, target;
    cin >> n;
    int *a = new int[n];
    for(int i = 0; i < n; i++){
        cin >> a[i];
    }
    cin >> target;
    int ans = binarySearch(a, 0, n - 1, target);
    cout << ans << endl;
    return 0;
}