35. 搜索插入位置
思路:设置一个指针idx。
遍历数组,当数组元素不等于target且小于target时,idx = idx + 1.
最后返回idx即可。
class Solution {
public int searchInsert(int[] nums, int target) {
if (nums.length == 0) return 0;
int idx = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] != target && target > nums[i]) {
idx++;
}
}
return idx;
}
}
思路:二分查找
先设定左侧下标 left 和右侧下标 right,再计算中间下标 mid = left + (right - left) / 2
每次根据 nums[mid] 和 target 之间的大小进行判断:
(1)nums[mid] < target 则 left 右移,即left = mid + 1;
(2)nums[mid] > target 则 right 左移,即right = mid - 1;
(3)相等则直接返回下标;
查找结束如果没有相等值则返回 left,该值为插入位置。
如图所示:
class Solution {
public int searchInsert(int[] nums, int target) {
if (nums.length == 0) return 0;
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
} else {
return mid;
}
}
return left;
}
}