0. 注意点
- 注意计算mid时的整形溢出
- 边界值判断
- 数组必须是有序的
1. 基本二分查找
局限性:无法查找target的左右侧边界
int binarySearch(vector<int>& nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
if (nums[mid] == target) {
return mid;
} else if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
}
}
return -1;
}
2. 寻找左侧边界的二分查找
// 寻找数组中最左边的目标[1, 2, 2, 2, 3]中index = 1的2
// [left, right]写法
int binarySearch(vector<int>& nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
if (nums[mid] == target) {
right = mid - 1; // 收缩右侧边界
} else if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
}
}
// 左侧边界的概念:nums小于target的个数有left个
// left 可能是0,代表比target小的有0个
// left 也可能是nums.size(),代表都比target小
if (left >= nums.size() || nums[left] != target) {
return -1;
}
return left;
}
3. 寻找右侧边界的二分查找
// 寻找数组中最右边的目标[1, 2, 2, 2, 3]中index = 3的2
// [left, right]写法
int binarySearch(vector<int>& nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
if (nums[mid] == target) {
left = mid + 1; // 收缩左侧边界
} else if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid - 1;
}
}
// 右侧边界的概念:nums小于等于target的个数有right个
// right 可能是nums.size() - 1,代表比target大的有0个
// right 也可能是-1,代表都比target大
if (right < 0 || nums[right] != target) {
return -1;
}
return right;
}
4. 二分查找函数
#include <algorithm>
vector<int> a = {1, 3, 3, 5, 7};
vector<int>::iterator it = lower_bound(a.begin(), a.end(), 3); // *it = 1, 查找有序区间a[i] >= k的最小指针
vector<int>::iterator it = upper_bound(a.begin(), a.end(), 3); // *it = 3, 查找有序区间a[i] > k的最小指针
bool ret = binary_search(a.begin(), a.end(), 3); // ret = true, 查找有序区间是否存在val。
// 注意查找区间:左闭右开,有序。适用于set\map\vector
// 函数lower_bound()在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置。
// 如果所有元素都小于val,则返回last的位置,且last的位置是越界的!
// 要计算下标pos,需要减去a.begin(),pos的可能值是[0, a.size()], 注意a.size()是越界的、代表所有值都比val小
int pos = lower_bound(a.begin(), a.end(), 3) - a.begin();
int b[5] = {1, 3, 3, 5, 7};
int pos = lower_bound(b, b + 5, 3) - a;
// 计算长度为n的有序数组a中k的个数
int num = upper_bound(a.begin(), a.end(), 3) - lower_bound(a.begin(), a.end(), 3);
num 为0,说明k的个数为0,不存在。 // 可作为判断k是否存在的依据,
// 将有序数组a中k所对应位置的元素改为k;
*lower_bound(a.begin(), a.end(), k) = k;
// 查找第一个最大值/最小值的地址,有相同大小的元素,找第一个
vector<int>::iterator it = max_element(a.begin(), a.end());
vector<int>::iterator it = min_element(a.begin(), a.end());
// 查找第一个最大值/最小值的下标
int pos = max_element(a.begin(), a.end()) - a.begin();
int pos = min_element(a.begin(), a.end()) - a.begin();
// 查找第一个最大值/最小值
int maxNum = *max_element(a.begin(), a.end());
int minNum = *min_element(a.begin(), a.end());
5. 二维数组的二分查找
// 编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
// 每行中的整数从左到右按升序排列。
// 每行的第一个整数大于前一行的最后一个整数。
class Solution {
public:
// 将二维数组转化为一维数组的方式
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m = matrix.size();
int n = matrix[0].size();
int left = 0;
int right = m * n - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
if (matrix[mid / n][mid % n] == target) {
return true;
} else if (matrix[mid / n][mid % n] < target) {
left = mid + 1;
} else if (matrix[mid / n][mid % n] > target) {
right = mid - 1;
}
}
return false;
}
};
6. 二分答案
给定一个非负整数数组 nums 和一个整数 m ,你需要将这个数组分成 m 个非空的连续子数组。
设计一个算法使得这 m 个子数组各自和的最大值最小。
// 二分查找左边界
class Solution {
public:
int splitArray(vector<int>& nums, int m) {
long long left = 0, right = 0;
for (int i = 0; i < nums.size(); i++) {
right += nums[i];
if (left < nums[i]) {
left = nums[i]; // 重点,确定起始值。不要认为是从0开始的。
}
}
while (left <= right) {
long long mid = (left + right) >> 1;
if (check(nums, mid, m)) { // 查找左边界,所以相等的时候,收缩右边界
right = mid - 1;
} else {
left = mid + 1;
}
}
return left;
}
bool check(vector<int>& nums, int x, int m) {
long long sum = 0;
int cnt = 1;
for (int i = 0; i < nums.size(); i++) {
if (sum + nums[i] > x) {
cnt++;
sum = nums[i];
} else {
sum += nums[i];
}
}
return cnt <= m;
}
};