Search Insert Position
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
My Code
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int size = nums.size();
int left_idx = 0;
int right_idx = size - 1;
int mid_idx, idx = -1;
while (left_idx <= right_idx)
{
mid_idx = (left_idx + right_idx) / 2;
if (nums[mid_idx] == target)
return mid_idx;
else if (nums[mid_idx] < target)
{
if (mid_idx == size - 1 || nums[mid_idx+1] > target)
return mid_idx + 1;
else
left_idx = mid_idx + 1;
}
else
{
if (mid_idx == 0 || nums[mid_idx-1] < target)
return mid_idx;
else
right_idx = mid_idx - 1;
}
}
}
};
Runtime: 8 ms