【计蒜客系列】挑战难题10：寻找插入位置

题目来源：计蒜客

给定一个已经升序排好序的数组，以及一个数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，以及其对应的数组A
，最后输入target

searchInsert(int A[], int n, int target)

样例1

输入：

3

1 3 5

2

输出：
1

#include <stdio.h>
#include <malloc.h>

int searchInsert(int A[], int n, int target){
int low = 0;
int high = n - 1;
int index; //未查找到时显示应该插入的索引
//使用二分查找target在不在A中
while(low <= high)
{
int mid = (low + high)/2;
if(A[mid] == target)
{
return mid;
}else if(A[mid] < target)
{
low = mid + 1;
index = low;
}else{
high = mid - 1;
index = high + 1;
}
}
return index;
}

int main(int argc, char **argv) {
int i,n,result,target;
scanf("%d",&n);
int *a = (int *)malloc(n * sizeof(int));
for(i=0;i<n;i++){
scanf("%d",&a[i]);
}
scanf("%d",&target);
result = searchInsert(a,n,target);
printf("%d",result);
return 0;
}