代码随想录学习
数组:
其他二分查找
34,35
题目:34
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
思路:
这道题需要找到一个数组里的目标值,但考虑到目标值并不仅仅有一个,所以需要同时考虑目标值的左右边界。可以寻找左右边界两次来确定最终位置。
寻找target在数组里的左右边界,有如下三种情况:
- 情况一:target 在数组范围的右边或者左边,例如数组{3, 4, 5},target为2,或者数组{3, 4, 5},target为6,此时应该返回{-1, -1}
- 情况二:target 在数组范围中,且数组中不存在target,例如数组{3,6,7}, target为5,此时应该返回{-1, -1}
- 情况三:target 在数组范围中,且数组中存在target,例如数组{3,6,7}, target为6,此时应该返回{1, 1}
落实到代码上,其和普通的二分查找区别不大,基本上都是先进行while loop 然后讨论
nums[middle] == target
在上式大于或小于时,还是按照二分查找进行即可。在等于时我们可以理解为找到了边界,并进行存储。之后继续进行while循环找是否还会有相同的边界。
我的关键实现:
int binSearch (int *nums, int numsSize, int target, int LR) {
int left = 0;
int right = numsSize -1;
int middle = 0;
int bound = -1;
while (left <= right) {
middle = left + (right - left) / 2;
if (nums[middle] > target) {
right = middle - 1;
}
else if (nums[middle] < target) {
left = middle + 1;
}
else {
bound = middle;
switch (LR) {
case 0:
right = middle - 1;
break;
case 1:
left = middle + 1;
break;
}
}
}
return bound;
}
代码通过评论中得到启发,如果按照随想录的格式写,在寻找边界时,进行了多次不必要的存储临时边界的操作。这个方法避免了存储临时边界。
题目:35
这个题目是很简单的二分查找变式,需要在找不到target的情况下把target插入到数组的特定位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
思路:
四种不同的插入可能
- 目标值在数组所有元素之前
- 目标值等于数组中某一个元素
- 目标值插入数组中的位置
- 目标值在数组所有元素之后
说白了!就是在基础的返回middle的基础上再返回一个不存在时的插入位置角标就好了
那么这个角标该怎么得到呢,我们考虑while的最后一步:我们都会得到一个左大右小的值。
考虑情况1:目标值在数组所有元素之前 eg:[1, 3, 5, 6] target = 0 最后left = 0,right = -1。此时0该放首位,则取left值。
考虑情况2: 目标值等于数组中某一个元素 这种情况直接二分查找。
考虑情况3: 目标值插入数组中的位置 eg:[1, 3, 5, 6] target = 4 最后left = 2,right = 1,此时target该放在5的位置下标为2,则还取left值。
考虑情况4: 目标值在所有的数组元素后 eg:[1, 3, 5, 6] target = 7最后left = 4,right = 3,此时target该放在6的右边就是下标为4的位置,则还取left值即可。
则我们得到了一个结论,如果二分查找能找到,则直接返回middle,否则返回left。
我的实现:
int searchInsert(int* nums, int numsSize, int target){
int left = 0;
int right = numsSize - 1;
int middle = 0;
while (left <= right) {
middle = (left + right) / 2;
if (nums[middle] < target) {
left = middle + 1;
}
else if (nums[middle] > target) {
right = middle - 1;
}
else if (nums[middle] == target) {
return middle;
}
}
return left;
}