给你一个下标从 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 <= 105
1 <= nums[i] <= nums.length
0 <= k <= nums.length
问题简要描述:返回最长等值子数组的长度
细节阐述:
- 哈希表 g 维护每个元素的下标列表
Java
class Solution {
public int longestEqualSubarray(List<Integer> nums, int k) {
int n = nums.size();
List<Integer>[] g = new List[n + 1];
Arrays.setAll(g, e -> new ArrayList<>());
for (int i = 0; i < n; i++) {
g[nums.get(i)].add(i);
}
int ans = 0;
for (List<Integer> ids : g) {
int l = 0;
for (int r = 0; r < ids.size(); r++) {
while ((ids.get(r) - ids.get(l)) - (r - l) > k) {
l++;
}
ans = Math.max(ans, r - l + 1);
}
}
return ans;
}
}
Python3
class Solution:
def longestEqualSubarray(self, nums: List[int], k: int) -> int:
g = defaultdict(list)
for i, v in enumerate(nums):
g[v].append(i)
ans = 0
for ids in g.values():
l = 0
for r in range(len(ids)):
while (ids[r] - ids[l]) - (r - l) > k:
l += 1
ans = max(ans, r - l + 1)
return ans
TypeScript
function longestEqualSubarray(nums: number[], k: number): number {
let n = nums.length;
let g: number[][] = Array.from({length: n + 1}, () => []);
for (let i = 0; i < n; i++) {
g[nums[i]].push(i);
}
let ans = 0;
for (const ids of g) {
let l = 0;
for (let r = 0; r < ids.length; r++) {
while ((ids[r] - ids[l]) - (r - l) > k) {
l++;
}
ans = Math.max(ans, r - l + 1);
}
}
return ans;
};
C++
class Solution {
public:
int longestEqualSubarray(vector<int>& nums, int k) {
int n = nums.size();
vector<vector<int>> g(n + 1);
for (int i = 0; i < n; i++){
g[nums[i]].emplace_back(i);
}
int ans = 0;
for (auto& ids : g) {
int l = 0;
for (int r = 0; r < ids.size(); r++){
while ((ids[r] - ids[l]) - (r - l) > k) {
l++;
}
ans = max(ans, r - l + 1);
}
}
return ans;
}
};