题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
思路
数组有序,二分查找
实现
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if (nums.size() == 0) {
return 0;
}
int start = 0;
int end = nums.size() - 1;
if (target > nums[end]) {
return (end + 1);
}
if (target < nums[start]) {
return 0;
}
while (start <= end) {
if ((end - start) == 1 && nums[end] != target && nums[start] != target) {
return end;
}
if (start == end && nums[start] != target) {
if (target < nums[start]) {
return (start - 1);
}
if (target > nums[start]) {
return (start + 1);
}
}
if (target == nums[start]) {
return start;
}
if (target == nums[end]) {
return end;
}
int mid = (start + end) / 2;
if (target == nums[mid]) {
return mid;
}
if (target > nums[mid]) {
start = mid;
} else {
end = mid;
}
}
return 0;
}
};