Every day a Leetcode
题目来源:2962. 统计最大元素出现至少 K 次的子数组
解法1:滑动窗口
算法如下:
- 设 mx = max(nums)。
- 右端点 right 从左到右遍历 nums。遍历到元素 x=nums[right] 时,如果 x=mx,就把计数器 count_mx 加一。
- 如果此时 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)。