Leetcode1224 最大相等频率

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

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

输入：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]，里面每个数字都出现了两次。
输入：nums = [1,1,1,2,2,2,3,3,3,4,4,4,5]
输出：13

解法一：
看到此题，首先想到的就是创建两个哈希表，一个用来存储数字出现的次数，一个用来存储频率相关的次数

class Solution {
    public int maxEqualFreq(int[] nums) {
        Map<Integer, Integer> freq = new HashMap<Integer, Integer>();
        Map<Integer, Integer> count = new HashMap<Integer, Integer>();
        int res = 0, maxFreq = 0;
        for (int i = 0; i < nums.length; i++) {
            if (count.getOrDefault(nums[i], 0) > 0) {
                freq.put(count.get(nums[i]), freq.get(count.get(nums[i])) - 1);
            }
            count.put(nums[i], count.getOrDefault(nums[i], 0) + 1);
            maxFreq = Math.max(maxFreq, count.get(nums[i]));
            freq.put(count.get(nums[i]), freq.getOrDefault(count.get(nums[i]), 0) + 1);
            boolean 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;
    }
}

解法二，由于数据范围较小，可以用数组作为简易哈希表

class Solution {
    public int maxEqualFreq(int[] nums) {
        int[] counts = new int[100001];
        int[] freqs = new int[100001];
        int ans = 0, maxFreq = 0;
        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];
            int count = counts[num];
            if (count > 0) {
                freqs[count]--;
            }
            counts[num] = ++count;
            freqs[count]++;
            maxFreq = Math.max(maxFreq, count);
            if (maxFreq == 1 
                    || maxFreq * freqs[maxFreq] == i
                    || maxFreq + (maxFreq - 1) * freqs[maxFreq - 1] == i + 1) {
                ans = Math.max(ans, i + 1);
            }
        }

        return ans;
    }
}

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/maximum-equal-frequency
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。