文章目录
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.
Example 1:
Input: [1,3,5,6], 5
Output: 2
Example 2:
Input: [1,3,5,6], 2
Output: 1
Example 3:
Input: [1,3,5,6], 7
Output: 4
Example 4:
Input: [1,3,5,6], 0
Output: 0
My Solution
一开始想尝试二分法,未果,还是用最快最熟悉的方法比较好。
此题坑在于target
小于nums
最小值和大于nums
最大值的情况,其他情况的案例都是返回大于等于target
的位置,唯有这两种情况一种返回大于target
位置的前一位置,一种返回最后一位置。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if (nums.at(0) > target)
return 0;
else if (nums.at(nums.size() - 1) < target)
return nums.size();
for (int i = 0; i < nums.size(); i++) {
if (nums.at(i) >= target)
return i;
}
return -1;
}
};