本文正在参加「Java主题月 - Java 刷题打卡」,详情查看 活动链接
今天520,你送花了吗?这个题目来计算制作 m 束花所需的最少天数。
题目描述
给你一个整数数组 bloomDay,以及两个整数 m 和 k 。
现需要制作 m 束花。制作花束时,需要使用花园中 相邻的 k 朵花 。
花园中有 n 朵花,第 i 朵花会在 bloomDay[i] 时盛开,恰好 可以用于 一束 花中。
请你返回从花园中摘 m 束花需要等待的最少的天数。如果不能摘到 m 束花则返回 -1 。
示例 1:
输入:bloomDay = [1,10,3,10,2], m = 3, k = 1
输出:3
解释:让我们一起观察这三天的花开过程,x 表示花开,而 _ 表示花还未开。
现在需要制作 3 束花,每束只需要 1 朵。
1 天后:[x, _, _, _, _] // 只能制作 1 束花
2 天后:[x, _, _, _, x] // 只能制作 2 束花
3 天后:[x, _, x, _, x] // 可以制作 3 束花,答案为 3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-number-of-days-to-make-m-bouquets
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析
- 这个题目有一定的难度,采用分治,拆分成两个小函数,一个函数用来判断某天是否能达到相应的花束,一个函数用来找天数最小值。
- 判断某天是否能达到相应的花束函数按照题意循环即可。
- 取最少天数,使用了二分查找的方法,通过遍历 bloomDay 数组,找到最小和最大的天数区间,然后通过二分查找,快速得到符合题目要求最小天数。
AC代码
public class DayCode {
public static void main(String[] args) {
int[] bloomDay = new int[]{1, 10, 3, 10, 2};
int m = 3;
int k = 1;
int ans = new DayCode().minDays(bloomDay, m, k);
System.out.println(ans);
}
public int minDays(int[] bloomDay, int m, int k) {
int n = bloomDay.length;
if (n < m * k) {
return -1;
}
int low = Integer.MAX_VALUE;
int high = 0;
for (int i = 0; i < n; i++) {
low = Math.min(low, bloomDay[i]);
high = Math.max(high, bloomDay[i]);
}
while (low < high) {
int days = low + (high - low) / 2;
if (help(bloomDay, days, m, k)) {
high = days;
} else {
low = days + 1;
}
}
return low;
}
private boolean help(int[] bloomDay, int days, int m, int k) {
int flowers = 0;
int bouquets = 0;
int n = bloomDay.length;
for (int i = 0; i < n && bouquets < m; i++) {
if (bloomDay[i] <= days) {
flowers++;
if (flowers == k) {
bouquets++;
flowers = 0;
}
} else {
flowers = 0;
}
}
return bouquets >= m;
}
}
提交测试:
总结
- 上述代码的时间复杂度是 O(n * log(n)), 空间复杂度是 O(1)
- 二分查找也称折半搜索,是用来在一个有序数组中查找某一元素的算法。
- 坚持每日一题,加油!