算法题解之二分专题
二分查找算法模板
二分模板一共有两个,分别适用于不同情况。
算法思路:假设目标在闭区间[l,r],每次将区间长度缩小一半,当l = r时,我们就找到了目标值。
版本1:
- 当我们将区间[l,r]划分成[l,mid]和[mid + l, r]时,其更新操作是r= mid或者l = mid + 1;计算mid时不需要加1
C++代码模板:
int bsearch_1(int l, int r) {
while (l < r) {
int mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
return l;
}
版本2:
- 当我们将区间[l , r]划分成[l, mid -1]和[mid , r]时,其更新操作是r = mid -1或者l = mid;此时为了防止死循环,计算mid时需要加1
C++代码模板:
int bsearch_2(int l, int r) {
while (l < r) {
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
二分的流程:
1、确定二分的边界
2、编写二分的代码框架
3、设计一个check(性质)
4、判读一下区间如何更新
5、如果更新方式写的是l = mid,r = mi-1,那么就在算mid的时候加上1
69. x 的平方根
【题目描述】:实现 int sqrt(int x) 函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
【C++代码】:
class Solution {
public:
int mySqrt(int x) {
int l = 0, r = x;
while(l < r){
int mid = l+(long long)r+1 >>1;
if(mid <=x/mid) l = mid;
else r = mid -1;
}
return r;
}
};
35. 搜索插入位置
【题目描述】:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。你可以假设数组中无重复元素。
【C++代码】:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(nums.empty()||nums.back() < target) return nums.size();
int l = 0, r = nums.size()-1;
while(l < r){
int mid = l + (long long)r >> 1;
if(nums[mid] >= target) r = mid;
else l = mid + 1;
}
return r;
}
};
34. 在排序数组中查找元素的第一个和最后一个位置
【题目描述】:给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。如果数组中不存在目标值 target,返回 [-1, -1]。
【C++代码】:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
if(nums.empty()) return {-1,-1};
int l = 0, r= nums.size()-1;
while(l < r){
int mid = l + r >> 1;
if(nums[mid]>= target) r = mid;
else l = mid +1;
}
if(nums[r]!=target) return {-1,-1};
int start = r;
l = 0,r = nums.size() -1;
while(l < r){
int mid = l + r + 1 >> 1;
if(nums[mid] <= target) l = mid;
else r = mid-1;
}
int end = r;
return {start ,end};
}
};
74. 搜索二维矩阵
【题目描述】:编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
- 每行中的整数从左到右按升序排列。
- 每行的第一个整数大于前一行的最后一个整数。
【C++代码】:
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrixs, int target) {
if (matrixs.empty() || matrixs[0].empty()) return false;
int m = matrixs.size(), n = matrixs.size();
int l = 0, r = n * m - 1;
while (l < r) {
int mid = l + r >> 1;
if (matrixs[mid / n][mid % n] >= target) r = mid;
else l = mid + 1;
}
if (matrixs[l / n][l % n] != target) return false;
return true;
}
};
153. 寻找旋转排序数组中的最小值
【题目描述】:假设按照升序排序的数组在预先未知的某个点上进行了旋转。例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] 。请找出其中最小的元素。
【C++代码】:
class Solution {
public:
int findMin(vector<int>& nums) {
if (nums.empty()) return 0;
int l = 0, r = nums.size() - 1;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] < nums.back())r = mid;
else l = mid + 1;
}
return nums[r];
}
};
33. 搜索旋转排序数组
153. 寻找旋转排序数组中的最小值
【题目描述】:升序排列的整数数组 nums 在预先未知的某个点上进行了旋转(例如, [0,1,2,4,5,6,7] 经旋转后可能变为 [4,5,6,7,0,1,2] )。请你在数组中搜索 target ,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
【C++代码】:
class Solution {
public:
int search(vector<int>& nums, int target) {
if (nums.empty()) return -1;
int l = 0, r = nums.size() - 1;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] <= nums.back()) r = mid;
else l = mid + 1;
}
if (target <= nums.back()) r = nums.size() - 1;
else l = 0, r--;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] >= target) r = mid;
else l = mid + 1;
}
if (nums[l] == target) return l;
return - 1;
}
};
278. 第一个错误的版本
【题目描述】:你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
【C++代码】:
class Solution {
public:
int firstBadVersion(int n) {
int l = 1,r = n;
while(l < r){
int mid = (long long)l + r >> 1;
if(isBadVersion(mid)) r = mid;
else l = mid + 1;
}
return r;
}
};
162. 寻找峰值
【题目描述】:峰值元素是指其值大于左右相邻值的元素。
给你一个输入数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。你可以假设 nums[-1] = nums[n] = -∞ 。
【C++代码】:
class Solution {
public:
int findPeakElement(vector<int>& nums) {
int l = 0,r = nums.size()-1;
while(l < r){
int mid = l + r >>1;
if(nums[mid] > nums[mid + 1]) r = mid;
else l = mid + 1;
}
return r;
}
};
287. 寻找重复数
【题目描述】:给定一个包含 n + 1 个整数的数组 nums ,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设 nums 只有 一个重复的整数 ,找出 这个重复的数 。
【C++代码】:
class Solution {
public:
int findDuplicate(vector<int>& nums) {
if (nums.empty()) return -1;
int l = 1, int r = nums.size() - 1;
while (l < r) {
int mid = l + r >> 1;
int cnt = 0;
for (auto x : nums)
if (x >= l && x <= mid)
cnt++;
if (cnt > mid - l + 1) r = mid;
else l = mid + 1;
}
return r;
}
};
275. H 指数 II
【题目描述】:给定一位研究者论文被引用次数的数组(被引用次数是非负整数),数组已经按照 升序排列 。编写一个方法,计算出研究者的 h 指数。
h 指数的定义: “h 代表“高引用次数”(high citations),一名科研人员的 h 指数是指他(她)的 (N 篇论文中)总共有 h 篇论文分别被引用了至少 h 次。(其余的 N - h 篇论文每篇被引用次数不多于 h 次。)"
【C++代码】:
class Solution {
public:
int hIndex(vector<int>& citations) {
int l =0, r = citations.size();
while(l < r){
int mid = l + (long long)r + 1 >> 1;
if(citations[citations.size()-mid] >=mid) l = mid;
else r = mid -1;
}
return r;
}
};
二分模板总结
二分的流程:
- 1、确定二分的边界
- 2、编写二分的代码框架
- 3、设计一个check(性质)
- 4、判断一下区间如何更新
- 5、如果更新方式写的是L= mid,r = mid-1,那么就在mid的时候加上1