力扣每日一题0818-1224. 最大相等频率

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第19天，点击查看活动详情

给你一个正整数数组 nums，请你帮忙从该数组中找出能满足下面要求的 最长 前缀，并返回该前缀的长度：

• 从前缀中 恰好删除一个 元素后，剩下每个数字的出现次数都相同。

如果删除这个元素后没有剩余元素存在，仍可认为每个数字都具有相同的出现次数（也就是 0 次）。

示例 1：

输入：nums = [2,2,1,1,5,3,3,5]
输出：7
解释：对于长度为 7 的子数组 [2,2,1,1,5,3,3]，如果我们从中删去 nums[4] = 5，就可以得到 [2,2,1,1,3,3]，里面每个数字都出现了两次。

示例 2：

输入：nums = [1,1,1,2,2,2,3,3,3,4,4,4,5]
输出：13

哈希表

使用哈希表 $\textit{count}$ 记录数 $x$ 出现的次数 $\textit{count}[x]$，$\textit{freq}$ 记录出现次数为 $f$ 的数的数目为 $\textit{freq}[f]$，$\textit{maxFreq}$ 表示最大出现次数。

依次遍历数组，假设当前访问的数为 $\textit{nums}[i]$，对应地更新 $\textit{count}$，$\textit{freq}$ 以及 $\textit{maxFreq}$。以 $\textit{nums}[i]$ 结尾的数组前缀符合要求的充要条件为满足以下三个条件之一：

• 最大出现次数 $\textit{maxFreq} = 1$ ：那么所有数的出现次数都是一次，随意删除一个数既可符合要求。
• 所有数的出现次数都是 $\textit{maxFreq}$ 或 $\textit{maxFreq} - 1$，并且最大出现次数的数只有一个：删除一个最大出现次数的数，那么所有数的出现次数都是 $\textit{maxFreq} - 1$。
• 除开一个数，其他所有数的出现次数都是 $\textit{maxFreq}$，并且该数的出现次数为 $1$：直接删除出现次数为 $1$ 的数，那么所有数的出现次数都是 $\textit{maxFreq}$。

var maxEqualFreq = function(nums) {
    const freq = new Map();
    const count = new Map();
    let res = 0, maxFreq = 0;
    for (let i = 0; i < nums.length; i++) {
        if (!count.has(nums[i])) {
            count.set(nums[i], 0);
        }
        if (count.get(nums[i]) > 0) {
            freq.set(count.get(nums[i]), freq.get(count.get(nums[i])) - 1);
        }
        count.set(nums[i], count.get(nums[i]) + 1);
        maxFreq = Math.max(maxFreq, count.get(nums[i]));
        if (!freq.has(count.get(nums[i]))) {
            freq.set(count.get(nums[i]), 0);
        }
        freq.set(count.get(nums[i]), freq.get(count.get(nums[i])) + 1);
        const ok = maxFreq === 1 ||
                freq.get(maxFreq) * maxFreq + freq.get(maxFreq - 1) * (maxFreq - 1) === i + 1 && freq.get(maxFreq) === 1 ||
                freq.get(maxFreq) * maxFreq + 1 === i + 1 && freq.get(1) === 1;
        if (ok) {
            res = Math.max(res, i + 1);
        }
    }
    return res;
};