1. 题目
给你一个整数数组 nums
和一个整数 k
。
一个元素 x
在数组中的 频率 指的是它在数组中的出现次数。
如果一个数组中所有元素的频率都 小于等于 k
,那么我们称这个数组是 好 数组。
请你返回 nums
中 最长好 子数组的长度。
子数组 指的是一个数组中一段连续非空的元素序列。
2. 示例
示例 1:
输入:nums = [1,2,3,1,2,3,1,2], k = 2 输出:6 解释:最长好子数组是 [1,2,3,1,2,3] ,值 1 ,2 和 3 在子数组中的频率都没有超过 k = 2 。[2,3,1,2,3,1] 和 [3,1,2,3,1,2] 也是好子数组。 最长好子数组的长度为 6 。
示例 2:
输入:nums = [1,2,1,2,1,2,1,2], k = 1 输出:2 解释:最长好子数组是 [1,2] ,值 1 和 2 在子数组中的频率都没有超过 k = 1 。[2,1] 也是好子数组。 最长好子数组的长度为 2 。
示例 3:
输入:nums = [5,5,5,5,5,5,5], k = 4 输出:4 解释:最长好子数组是 [5,5,5,5] ,值 5 在子数组中的频率没有超过 k = 4 。 最长好子数组的长度为 4 。
3. 思路
这道题目是一道滑动窗口类型的题目,当时比赛的时候一直没想起怎么移动,想着靠哈希表来存储每个数的第一个索引来移动窗口左边界,在比完后看到一个大佬的题解犹如醍醐灌顶,咱们可以在遍历的for循环里面套个循环来模拟移动窗口左边界,然后在每次遍历结束后更新最大值。下面是示例1的动画演示:
最多k个重复元素的最长子数组
4. 代码
class Solution:
def maxSubarrayLength(self, nums: List[int], k: int) -> int:
left = 0 # 初始窗口左边界
Max = 0 # 最大值
cnt = Counter() # 哈希表结构,比字典好用
for right,x in enumerate(nums): # 遍历数组,right为有边界索引,x为当前值
cnt[x] += 1 # 将当前数的次数+1
while cnt[x] > k: # 尝试移动窗口左边界
cnt[nums[left]] -= 1 # 在右移的过程将该数出现的次数-1
left += 1 # 不断向右移动窗口的左边界
Max = max(Max,right - left + 1) # 更新最大值
return Max # 返回结果