2831.找出最长等值子数组
给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。
如果子数组中所有元素都相等,则认为子数组是一个 等值子数组 。注意,空数组是 等值子数组 。
从 nums 中删除最多 k 个元素后,返回可能的最长等值子数组的长度。
子数组 是数组中一个连续且可能为空的元素序列。
示例 1:
输入:nums = [1,3,2,3,1,3], k = 3
输出:3
解释:最优的方案是删除下标 2 和下标 4 的元素。
删除后,nums 等于 [1, 3, 3, 3] 。
最长等值子数组从 i = 1 开始到 j = 3 结束,长度等于 3 。
可以证明无法创建更长的等值子数组。
示例 2:
输入:nums = [1,1,2,2,1,1], k = 2
输出:4
解释:最优的方案是删除下标 2 和下标 3 的元素。
删除后,nums 等于 [1, 1, 1, 1] 。
数组自身就是等值子数组,长度等于 4 。
可以证明无法创建更长的等值子数组。
提示:
1 <= nums.length <= 10^5
1 <= nums[i] <= nums.length
0 <= k <= nums.length
点击查看代码
class Solution {
public int longestEqualSubarray(List<Integer> nums, int k) {
int n = nums.size();
List<Integer>[] res = new ArrayList[n+1];
Arrays.setAll(res, i -> new ArrayList<>());
for(int i=0; i<n; i++){
res[nums.get(i)].add(i);
}
int max_len = 0;
for(List<Integer> l:res){
if(l.size() < max_len)
continue;
int left = 0;
for(int right=0; right<l.size(); right++){
while(l.get(right)-l.get(left)-right+left > k){
left++;
}
max_len = Math.max(max_len,right-left+1);
}
}
return max_len;
}
}

浙公网安备 33010602011771号