Leetcode2962. 统计最大元素出现至少 K 次的子数组

Every day a Leetcode

题目来源：2962. 统计最大元素出现至少 K 次的子数组

解法1：滑动窗口

算法如下：

1. 设 mx = max⁡(nums)。
2. 右端点 right 从左到右遍历 nums。遍历到元素 x=nums[right] 时，如果 x=mx，就把计数器 count_mx 加一。
3. 如果此时 count_mx>=k，则不断右移左指针 left，直到窗口内的 mx 的出现次数小于 k 为止。此时，对于右端点为 right 且左端点小于 left 的子数组，mx 的出现次数都至少为 k，把答案增加 left。

代码：

/*
 * @lc app=leetcode.cn id=2962 lang=cpp
 *
 * [2962] 统计最大元素出现至少 K 次的子数组
 */

// @lc code=start

// 滑动窗口

class Solution
{
public:
    long long countSubarrays(vector<int> &nums, int k)
    {
        int mx = *max_element(nums.begin(), nums.end());
        long long ans = 0;
        int count_mx = 0, left = 0;
        for (int right = 0; right < nums.size(); right++)
        {
            if (nums[right] == mx)
                count_mx++;
            while (count_mx >= k)
            {
                if (nums[left] == mx)
                    count_mx--;
                left++;
            }
            // [0, right],...,[left-1, right] 是 mx 至少出现 k 次的子数组
            ans += left;
        }
        return ans;
    }
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 是数组 nums 的长度。

空间复杂度：O(1)。