本来不想记录却又要记录的一条算法(被面试官狠狠羞辱回来怒刷算法却发现连二分查找都不会的彩笔)
P704 二分查找
题目
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例 1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例 2:
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
提示:
你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。
败绩
看败绩就知道为什么要记录。
题解
class Solution {
public int search(int[] nums, int target) {
int l=0,r=nums.length-1,mid=l;
while(l<=r){
mid = (l + r) / 2;
if (target < nums[mid]) {
r = mid-1;
} else if (target > nums[mid]) {
l = mid+1;
} else {
return mid;
}
}
return -1;
}
}
分析
主要是要记住mid一定要向中间逼近,即target<nums[mid]时,mid-1,target>nums[mid]时,mid+1。
换一个角度,
还未进入循环时,边界为0和n,0和n都有可能为target,
进入循环时,若nums[mid]!=target,下一次的左或右边界应不包含mid,
这样才符合循环子的规律(即二分查找每一次循环的前置条件和做的事情都一致)
还有一件事,二分查找的结束条件一般是left>right
P278 第一个错误版本
题目
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
示例 1:
输入:n = 5, bad = 4
输出:4
解释:
调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true
所以,4 是第一个错误的版本。
示例 2:
输入:n = 1, bad = 1
输出:1
提示:
1 <= bad <= n <=231-1
败绩
败绩+10086
题解
/* The isBadVersion API is defined in the parent class VersionControl.
boolean isBadVersion(int version); */
public class Solution extends VersionControl {
public int firstBadVersion(int n) {
int l = 1, r = n, mid = -1;
while (l < r) {
mid =l+ (r - l) / 2;
if (isBadVersion(mid)) {
r = mid;
} else {
l = mid+1;
}
}
return l;
}
}
分析
还是一样的道理,二分查找每一次循环的前置条件和做的事情都一致
每一次查找的前置条件第一个最坏版本一定在最左边界的右侧(不包含左边界),一定在右边界的左侧(包含右边界,这一特性导致本题查找为什么mid不用减一和为什么查找结束条件为left>=right)。
还有一个小细节,题目给的n大于int的范围,使用l+ (r - l) / 2求mid或long防止溢出导致错误
P35 搜索插入位置
题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为 无重复元素 的 升序 排列数组
-104 <= target <= 104
题解
class Solution {
public int searchInsert(int[] nums, int target) {
int l=0,r=nums.length-1,mid=-1;
while(l<=r){
mid=(l+r)/2;
if(target<nums[mid]){
r=mid-1;
}
else if(target>nums[mid]){
l=mid+1;
}
else{
return mid;
}
}
return target>nums[mid] ?mid+1:mid;
}
}
分析
略(懒得写了改日一定)
和之前类似,注意返回的值应为插入位置
总结
注意每次查找的边界问题和结束条件,还是蛮重要的
二分查找这种简单的经典题目都不会写了,可够彩笔的……
总而言之,每天多努力写一点点吧
——算法——