原题链接:
http://oj.leetcode.com/problems/search-insert-position/
二分查找是一个非常经典的方法,不过一般在面试中很少直接考二分查找,会考一些变体,例如 Search in Rotated Sorted Array , Search for a Range和 Search a 2D Matrix ,思路其实是类似的,稍微变体一下即可,有兴趣可以练习一下哈。
这道题比较简单,就是二分查找。思路就是每次取中间,如果等于目标即返回,否则根据大小关系切去一半。因此算法复杂度是O(logn),空间复杂度O(1)。代码如下:
public int searchInsert(int[] A, int target) {
if(A == null || A.length == 0)
{
return 0;
}
int l = 0;
int r = A.length-1;
while(l<=r)
{
int mid = (l+r)/2;
if(A[mid]==target)
return mid;
if(A[mid]<target)
l = mid+1;
else
r = mid-1;
}
return l;
}
注意以上实现方式有一个好处,就是当循环结束时,如果没有找到目标元素,那么l一定停在恰好比目标大的index上,r一定停在恰好比目标小的index上,所以个人比较推荐这种实现方式。
二分查找是一个非常经典的方法,不过一般在面试中很少直接考二分查找,会考一些变体,例如 Search in Rotated Sorted Array , Search for a Range和 Search a 2D Matrix ,思路其实是类似的,稍微变体一下即可,有兴趣可以练习一下哈。