持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第10天，点击查看活动详情

最近一直在力扣刷题，也逐渐对各类题型有了自己的理解，所谓见招拆招，将自己的浅显经验分享一下，帮助更多在编程路上的朋友们。

本题为力扣杯秋赛战队赛第三题

美观的花束

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

提示：

• 1 <= flowers.length <= 10^5
• 1 <= flowers[i] <= 10^5
• 1 <= cnt <= 10^5

思路

很容易想到的是根据长度作为第一层遍历，不同起始位置的区间作为第二层遍历，但此时时间复杂度为O(n²)，会超时，故不能一个个区间的去找。

题目要求的是符合条件的区间个数，如果长区间符合条件，那么里面的每个小区间都符合，如果区间长度为n，则符合条件的区间数为1 + 2 + ... + n。

所以我们每次只需要查找符合条件的最长区间即可。

题解

class Solution {
    final int MOD = (int) 1e9 + 7;
    public int beautifulBouquet(int[] flowers, int cnt) {
        int n = flowers.length;
        long ans = 0;
        int max = flowers[0];
        for(int f: flowers) {
            max = max < f? f: max;
        }
        int[] count = new int[max + 1];
        for(int left = 0, right = 0; left < n; left++) {
            while(right < n && count[flowers[right]] + 1 <= cnt) {
                count[flowers[right++]]++;
            }
            ans += right - left;
            ans %= MOD;
            count[flowers[left]]--;
        }
        return (int) ans;
    }
}