基本的二分搜索
int binarySearch(int[] nums, int target) {
int left = 0;
int right = nums.length - 1; // 注意 此时两端闭区间
while(left <= right) {
int mid = left + (right - left) / 2;
if(nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1; // 注意 mid已经搜索过,更新两端闭区间
else if (nums[mid] > target)
right = mid - 1; // 注意 mid已经搜索过,更新两端闭区间
}
return -1;
}
查找左侧边界的二分
int left_bound(int[] nums, int target) {
int left = 0;
int right = nums.length; // 注意 左闭右开区间
while (left < right) { // 注意
int mid = left + (right - left) / 2;
if (nums[mid] == target) { //查找左侧,更右侧
right = mid;
} else if (nums[mid] < target) {
left = mid + 1; //更新左闭区间
} else if (nums[mid] > target) {
right = mid; // 注意
}
}
return left;
}
查找右侧边界的二分
int right_bound(int[] nums, int target) {
int left = 0, right = nums.length;
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) { //查找右侧,更左侧
left = mid + 1; // 注意
} else if (nums[mid] < target) {
left = mid + 1;
} else if (nums[mid] > target) {
right = mid;
}
}
return left - 1; // 注意
}
二分搜索:
从题⽬中抽象出⼀个自变量 x,⼀个关于 x 的函数 f(x),以及⼀个⽬标值 target。
x, f(x), target 还要满⾜以下条件:
- f(x) 必须是在 x 上的单调函数(单调增单调减都可以)。
- 题⽬是让你计算满⾜约束条件 f(x) == target 时的 x 的值。
// 函数 f 是关于⾃变量 x 的单调函数
int f(int x) {
// ...
}
// 主函数,在 f(x) == target 的约束下求 x 的最值
int solution(int[] nums, int target) {
if (nums.length == 0) return -1;
// 问⾃⼰:⾃变量 x 的最⼩值是多少?
int left = ...;
// 问⾃⼰:⾃变量 x 的最⼤值是多少?
int right = ...+1;
while (left < right) {
int mid = left + (right - left) / 2;
if (f(mid) == target) {
// 问⾃⼰:题⽬是求左边界还是右边界?
// ...
} else if (f(mid) < target) {
// 问⾃⼰:怎么让 f(x) ⼤⼀点?
// ...
} else if (f(mid) > target) {
// 问⾃⼰:怎么让 f(x) ⼩⼀点?
// ...
}
}
return left;
}
二分搜索示例:leetcode-1011
传送带上的包裹必须在 days 天内从一个港口运送到另一个港口。
传送带上的第 i 个包裹的重量为 weights[i]。每一天,我们都会按给出重量(weights)的顺序往传送带上装载包裹。我们装载的重量不会超过船的最大运载重量。
返回能在 days 天内将传送带上的所有包裹送达的船的最低运载能力。
示例 1:
输入:weights = [1,2,3,4,5,6,7,8,9,10], days = 5
输出:15
解释:
船舶最低载重 15 就能够在 5 天内送达所有包裹,如下所示:
第 1 天:1, 2, 3, 4, 5
第 2 天:6, 7
第 3 天:8
第 4 天:9
第 5 天:10
请注意,货物必须按照给定的顺序装运,因此使用载重能力为 14 的船舶并将包装分成 (2, 3, 4, 5), (1, 6, 7), (8), (9), (10) 是不允许的。
示例 2:
输入:weights = [3,2,2,4,1,4], days = 3
输出:6
解释:
船舶最低载重 6 就能够在 3 天内送达所有包裹,如下所示:
第 1 天:3, 2
第 2 天:2, 4
第 3 天:1, 4
示例 3:
输入:weights = [1,2,3,1,1], days = 4
输出:3
解释:
第 1 天:1
第 2 天:2
第 3 天:3
第 4 天:1, 1
//leetcode submit region begin(Prohibit modification and deletion)
/**
* f(x) 运完所有货物需要的天数
* x 船装载的重量
* left 最小装载重量
* right最大装载量
* mid值越大,midDay越小
*/
class Solution {
public int shipWithinDays(int[] weights, int days) {
int left = 0;
int right = 0;
for (int i = 0; i < weights.length; i++) {
left = left < weights[i] ? weights[i] : left;
right += weights[i];
}
while (left < right) {
int mid = left + (right - left) / 2;
int midDay = ff(weights, mid);
if (midDay <= days) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
}
int ff(int[] weights, int mid) {
int days = 0;
int sum = 0;
for (int i = 0; i < weights.length; ) {
if (sum + weights[i] > mid) {
days++;
sum = 0;
continue;
}
sum += weights[i];
i++;
}
return ++days;
}
}
//leetcode submit region end(Prohibit modification and deletion)