找出最长等值子数组(Lc2831)——哈希表+双指针

给你一个下标从 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

问题简要描述：返回最长等值子数组的长度 

细节阐述:

1. 哈希表 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;    
    }
};