注:本文参考LeetCode LCP 12. 小张刷题计划(二分查找)
原题
为了提高自己的代码能力,小张制定了 LeetCode 刷题计划,他选中了 LeetCode 题库中的 n 道题,编号从 0 到 n-1,并计划在 m 天内按照题目
编号顺序刷完所有的题目(注意,小张不能用多天完成同一题)。
在小张刷题计划中,小张需要用 time[i] 的时间完成编号 i 的题目。此外,小张还可以使用场外求助功能,通过询问他的好朋友小杨题目的解法,
可以省去该题的做题时间。为了防止“小张刷题计划”变成“小杨刷题计划”,小张每天最多使用一次求助。
我们定义 m 天中做题时间最多的一天耗时为 T(小杨完成的题目不计入做题总时间)。请你帮小张求出最小的 T是多少。
示例 1:
输入:time = [1,2,3,3], m = 2
输出:3
解释:第一天小张完成前三题,其中第三题找小杨帮忙;第二天完成第四题,并且找小杨帮忙。这样做题时间最多的一天花费了 3 的时间,并且这个值是最小的。
示例 2:
输入:time = [999,999,999], m = 4
输出:0
解释:在前三天中,小张每天求助小杨一次,这样他可以在三天内完成所有的题目并不花任何时间。
限制:
1 <= time.length <= 10^5
1 <= time[i] <= 10000
1 <= m <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/xiao-zhang-shua-ti-ji-hua
思路
Ⅰ. 这是一类题
最大化最小值,最大化平均值,最小化最大值…
这类题主要的方法就是二分答案/搜索。
Ⅱ. 类似的题目
Ⅲ. 代码
- C++代码
class Solution {
private:
bool check(int mid, int m, vector<int> &time) {
// 所用天数, 总用时, 题耗时最多
// 这里用了贪心的思想, 耗时最多的那个题目, 就场外求助
int days = 1, total = 0, maxCost = 0;
// 每天的场外求助是否被使用
bool use = true;
int size = time.size();
for (int i = 0; i < size; ++i) {
// 维护那个耗时最长的题的花费时间
maxCost = max(maxCost, time[i]);
// 总用时
total += time[i];
// 如果大于我给定的时间了
if (total > mid) {
// 如果有场外求助, 说明我今天还可以多解决一道题
if (use) {
total -= maxCost;
use = false;
} else {
// 如果没有场外求助, 开启下一天, 所有信息清空
++days;
total = 0;
maxCost = 0;
--i;
use = true;
}
}
// 如果天数大于m了, 说明一天只有mid的时间不够, 返回false
if (days > m) {
return false;
}
}
return true;
}
public:
int minTime(vector<int>& time, int m) {
int left = 0, right = 0;
// 左边界=>0, 右边界=>所有题的总时间
for (auto &it : time) {
right += it;
}
while (left <= right) {
int mid = left + (right - left) / 2;
// 每天用mid的时间可以完成吗?
if (check(mid, m, time)) {
// 如果可以完成, 说明时间还可以减少
right = mid - 1;
} else {
// 如果不能完成, 说明时间不够
left = mid + 1;
}
}
return left;
}
};
- Python代码
class Solution:
def check(self, mid, m, time):
use = True
days, total, maxCost = 1, 0, 0
size = len(time)
i = 0
while i < size:
maxCost = max(maxCost, time[i])
total += time[i]
if total > mid:
if use:
total -= maxCost
use = False
else:
days += 1
i -= 1
use, total, maxCost = True, 0, 0
i += 1
if days > m:
return False
return True
def minTime(self, time: List[int], m: int) -> int:
left, right = 0, sum(time)
while left <= right:
mid = (left + right) // 2
if self.check(mid, m, time):
right = mid - 1
else:
left = mid + 1
return left
特别注意:Python中在 for in range(size)
中修改i的值是没有作用的,也强烈建议不要这样子做
原因是因为 for
循环每次结束 i
都会被重新赋值。
Why can’t you modify lists through “for in” loops in Python?