力扣嘉年华的花店中从左至右摆放了一排鲜花,记录于整型一维矩阵 flowers 中每个数字表示该位置所种鲜花的品种编号。你可以选择一段区间的鲜花做成插花,且不能丢弃。 在你选择的插花中,如果每一品种的鲜花数量都不超过 cnt 朵,那么我们认为这束插花是 「美观的」。
例如:[5,5,5,6,6] 中品种为 5 的花有 3 朵, 品种为 6 的花有 2 朵,每一品种 的数量均不超过 3
请返回在这一排鲜花中,共有多少种可选择的区间,使得插花是「美观的」。
注意:
答案需要以 1e9 + 7 (1000000007) 为底取模,如:计算初始结果为:1000000008,请返回 1
示例 1:
输入:flowers = [1,2,3,2], cnt = 1
输出:8
解释:相同的鲜花不超过 1 朵,共有 8 种花束是美观的;
长度为 1 的区间 [1]、[2]、[3]、[2] 均满足条件,共 4 种可选择区间
长度为 2 的区间 [1,2]、[2,3]、[3,2] 均满足条件,共 3 种可选择区间
长度为 3 的区间 [1,2,3] 满足条件,共 1 种可选择区间。
区间 [2,3,2],[1,2,3,2] 都包含了 2 朵鲜花 2 ,不满足条件。
返回总数 4+3+1 = 8
示例 2:
输入:flowers = [5,3,3,3], cnt = 2
输出:8
提示:
1 <= flowers.length <= 10^5
1 <= flowers[i] <= 10^5
1 <= cnt <= 10^5
解题思路:
使用 left 表示滑窗左边界,使用 right 表示滑窗右边界
可以使用 Map 对滑窗内的元素个数进行统计
当满足条件时扩展右边界,不满足条件时缩短左边界
遍历数组,计算以每个元素作为右边界(同时为子区间右边界)时,窗口内有多少个满足题意的子区间
可发现每次操作后,恰好有 right - left + 1 个子区间满足第 4 点
以示例 1 为例,使用滑动窗口可取得 [1], [1,2], [1,2,3], [3,2] 四个窗口
窗口 [1] 包含 [1] 1 个子区间满足题意
窗口 [1,2] 包含 [1,2], [2] 2 个子区间满足题意
窗口 [1,2,3] 包含 [1,2,3], [2,3], [3] 3 个子区间满足题意
窗口 [3,2] 包含 [3,2], [2] 2 个子区间满足题意
求和即可得到总数
只需要遍历一次数组,时间复杂度为 O(n)O(n)
需要使用一个 Map 对滑窗内元素进行统计,空间复杂度为 O(n)O(n)
* @param {number[]} flowers
* @param {number} cnt
* @return {number}
*/
var beautifulBouquet = function (flowers, cnt) {
const freq = [];
let ans = 0;
let left = 0;
for (let right = 0; right < flowers.length; right++) {
const n = flowers[right];
// 计数
freq[n] = -~freq[n];
// 当元素个数超过 cnt,缩短左边界
while (freq[n] > cnt) {
freq[flowers[left++]]--;
}
ans = (ans + right - left + 1) % 1000000007;
}
return ans;
};