美观的花束

150 阅读1分钟

力扣嘉年华的花店中从左至右摆放了一排鲜花,记录于整型一维矩阵 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;
};