1224. 最大相等频率

https://leetcode.cn/problems/maximum-equal-frequency/

给你一个正整数数组 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

提示：

2 <= nums.length <= 10^5
1 <= nums[i] <= 10^5

解题思路

见注释

代码

class Solution {
public:
    /*
    * ①最大出现次数 maxFreq = 1：那么所有数的出现次数都是一次，随意删除一个数既可符合要求。
    * ②所有数的出现次数都是maxFreq 或 maxFreq - 1，
    *   并且出现maxFreq次数的数只有一个：删除一个最大出现次数的数，
    *   那么所有数的出现次数都是 maxFreq - 1。
    * ③除开一个数，其他所有数的出现次数都是 maxFreq，
    *   并且该数的出现次数为 1：直接删除出现次数为 1 的数，
    *   那么所有数的出现次数都是 maxFreq。
    */
    int maxEqualFreq(vector<int>& nums) {
        unordered_map<int, int> count; // 某数字的个数，计数表
        unordered_map<int, int> freq; // 出现次数的数字个数，频次表
        int res = 0, maxFreq = 0; // 结果 最大出现次数
        for(int i = 0; i < nums.size(); i++){
            int num = nums[i];
            if(count[num] > 0) freq[count[num]]--; // 若此数字曾经出现过，则再一次出现，需要更新频次表 
            count[num]++;
            freq[count[num]]++;
            maxFreq = max(maxFreq, count[num]);
            bool ok = (maxFreq == 1) || // 最大出现频次为1，说明当前出现的所有数字都出现一次，随便删一个都行
                (freq[maxFreq] == 1 && i+1 == (freq[maxFreq]*maxFreq + freq[maxFreq-1]*(maxFreq-1))) || // 出现maxFreq次数的数只有一个：删除一个最大出现次数的数即可，同时所有数都只能是出现maxFreq和maxFreq-1次数
                (freq[maxFreq]*maxFreq == i); // 只有一个数的出现次数为1，其他所有数的出现次数都是 maxFreq：直接删除出现次数为 1 的数
            if(ok) res = max(res, i+1);
        }
        return res;
    }
};