这应该是比较简单的一道题了,代码如下
int searchInsert(int* nums, int numsSize, int target)
{
int i = 0;
if(numsSize <= 0)
{
return 0;
}
for(;i < numsSize;i ++)
{
if(nums[i] == target)
{
return i;
}
if(i == numsSize - 1)
{
if(nums[i] < target)
{
return numsSize;
}
}
if(nums[i] < target && nums[i+1] > target)
{
return i + 1;
}
}
return 0;
}
因为这个题目是查找一个有序数组中元素的位置,考虑可以使用二分法。修改代码后再次提交
int searchInsert(int* nums, int numsSize, int target)
{
int low = 0;
int high = numsSize - 1;
if(numsSize < 0)
{
return 0;
}
while(low <= high)
{
int middle = (low + high) / 2;
if(target == nums[middle])
{
return middle;
}
else if(target < nums[middle])
{
high = middle - 1;
}
else if(target > nums[middle])
{
low = middle + 1;
}
}
return low;
}
注意以上实现方式有一个好处,就是当循环结束时,如果没有找到目标元素,那么low一定停在恰好比目标大的index上,high一定停在恰好比目标小的index上。二分查找是一个非常经典的方法,不过一般在面试中很少直接考二分查找,会考一些变体。
参考资料
1 https://www.cnblogs.com/ccsccs/articles/4215628.html
2 《C语言大学教程(第八版)》