问题描述:
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0
分析问题:
给定一个升序的数组,然后给定一个target数,如果target在这个当前数组中,返回这个数组中的位置。如果不在这个数组中,找到这个数组应该插入的位置。可以用二分查找,找出问题的解决方案。
代码实现:
public int searchInsert(int[] nums, int target) {
int left = 0;
int right = nums.length - 1;
while (left <= right) {
int mid = (left + right) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
} else if (nums[mid] == target) {
return mid;
}
}
return left;
}
总结
二分查找如果没有找到,最终left将会定位到比当前target大一个的数的位置,right定位到比target小一个的位置上。抓住这点,这个题目就比较简单了。