力扣 1482. 制作 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

示例2：

输入：bloomDay = [1,10,3,10,2], m = 3, k = 2
输出：-1
解释：要制作 3 束花，每束需要 2 朵花，也就是一共需要 6 朵花。而花园中只有 5 朵花，无法满足制作要求，返回 -1 。

示例3：

输入：bloomDay = [7,7,7,7,12,7,7], m = 2, k = 3
输出：12
解释：要制作 2 束花，每束需要 3 朵。
花园在 7 天后和 12 天后的情况如下：
7 天后：[x, x, x, x, _, x, x]
可以用前 3 朵盛开的花制作第一束花。但不能使用后 3 朵盛开的花，因为它们不相邻。
12 天后：[x, x, x, x, x, x, x]
显然，我们可以用不同的方式制作两束花。

示例4：

输入：bloomDay = [1000000000,1000000000], m = 1, k = 1
输出：1000000000
解释：需要等 1000000000 天才能采到花来制作花束

示例5：

输入：bloomDay = [1,10,2,9,3,8,4,7,5,6], m = 4, k = 2
输出：9

提示：

• bloomDay.length == n
• 1 <= n <= 10⁵
• 1 <= bloomDay[i] <= 10⁹
• 1 <= m <= 10⁶
• 1 <= k <= n

问题思路

思路1：二分查找

首先，当需要的总花数 k × m 小于 bloomDay 的长度 n 时，无法制作出指定数量的花束，返回 -1。否则一定可以制作出指定数量的花束。

分析题意可知，一定存在一个 mday 值，使得当天数小于 mday 时，无法完成制作；当天数 day 大于等于 mday 时，一定能完成制作。

因此，可以采用二分查找的方法寻找这个 mday 值，初始左边界为 mindays 数组的最小值 l，右边界为 mindays 数组的最大值 r。同时实现一个辅助函数 canDo ，用来判断天数为 day 时，能否完成制作。

实现代码如下，空间复杂度O(1)，时间复杂度O(n log(r - l))。运行时间 19 ms。

class Solution {
	public int minDays(int[] bloomDay, int m, int k) {
		int n = bloomDay.length;
		if (k * m > n) {
			return -1;
		}
		int l = Integer.MAX_VALUE;
		int r = 0;
		for (int i : bloomDay) {
			l = Math.min(l, i);
			r = Math.max(r, i);
		}
		while (l < r) {
			int day = l + r >> 1;
			if (canDo(day, bloomDay, m, k)) {
				r = day;
			} else {
				l = day + 1;
			}
		}
		return l;
	}

	private boolean canDo(int day, int[] bloomDay, int m, int k) {
		int n = bloomDay.length;
		int flower = 0;
		int bonquet = 0;
		for (int i = 0; i < n && bonquet < m; i++) {
			if (bloomDay[i] <= day) {
				flower++;
				if (flower == k) {
					flower = 0;
					bonquet++;
				}
			} else {
				flower = 0;
			}
		}
		return bonquet == m;
	}
}