哈希表题目：最大相等频率

题目

标题和出处

标题：最大相等频率

出处：1224. 最大相等频率

难度

8 级

题目描述

要求

给你一个正整数数组 $\text{nums}$，请你从该数组中找出符合下面要求的最长前缀的长度：从前缀中恰好删除一个元素后，剩下每个数字的出现次数都相同。

如果删除一个元素后没有剩余元素，仍可认为每个数字都具有相同的出现次数（$\text{0}$ 次）。

示例

示例 1：

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

示例 2：

输入：$\text{nums = [1,1,1,2,2,2,3,3,3,4,4,4,5]}$
输出：$\text{13}$

数据范围

• $\text{2}\le \text{nums.length}\le {\text{10}}^{\text{5}}$
• $\text{1}\le \text{nums[i]}\le {\text{10}}^{\text{5}}$

解法

思路和算法

为了在数组 $\text{nums}$ 中找到符合要求的最长前缀的长度，需要遍历数组 $\text{nums}$ 的每一个前缀并分别判断是否符合要求。假设数组 $\text{nums}$ 的长度是 $n$，由于 $n$ 的最大值是 $10^5$，因此需要使用 $O(n)$ 的解法，每次判断前缀是否符合要求的时间复杂度应该是 $O(1)$。

假设一个符合要求的前缀的长度是 $m$，则在该前缀中存在一个元素，删除该元素之后剩下的 $m-1$ 个元素满足有 $x$ 个不同的元素且这 $x$ 个不同的元素各出现 $y$ 次，即 $x\times y=m-1$。

当 $m=1$ 时，有 $y=0$。当 $m>1$ 时，$m-1>0$，因此 $x$ 和 $y$ 都是正整数且都在范围 $[1,m-1]$ 内。可能的情况有如下 $3$ 种。

1. $x=1$，$y=m-1$，此时 $m-1$ 个元素全部相同。

2. $x=m-1$，$y=1$，此时 $m-1$ 个元素各不相同。

3. $1<x<m-1$，$y=\frac{m-1}{x}$，此时 $m-1$ 个元素不完全相同且每个元素的出现频率都大于 $1$。

对于上述 $3$ 种情况，分别考虑被删除的元素的可能取值，以及对应的 $m$ 个元素中的不同元素个数和每个元素的出现频率。

1. $x=1$，$y=m-1$：

   • 如果被删除的元素等于其余的 $x$ 个元素，则 $m$ 个元素中有 $1$ 个不同元素，出现 $m$ 次；

   • 如果被删除的元素不等于其余的 $x$ 个元素，则 $m$ 个元素中有 $2$ 个不同元素，分别出现 $m-1$ 次和 $1$ 次。

2. $x=m-1$，$y=1$：

   • 如果被删除的元素等于其余的 $x$ 个元素中的一个，则 $m$ 个元素中有 $m-1$ 个不同元素，其中一个元素出现 $2$ 次，其余元素各出现 $1$ 次；

   • 如果被删除的元素和其余的 $x$ 个元素都不相等，则 $m$ 个元素中有 $m$ 个不同元素，各出现 $1$ 次。

3. $1<x<m-1$，$y=\frac{m-1}{x}$：

   • 如果被删除的元素等于其余的 $x$ 个元素中的一个，则 $m$ 个元素中有 $x$ 个不同元素，其中一个元素出现 $y+1$ 次，其余元素各出现 $y$ 次；

   • 如果被删除的元素和其余的 $x$ 个元素都不相等，则 $m$ 个元素中有 $x+1$ 个不同元素，其中一个元素出现 $1$ 次，其余元素各出现 $y$ 次。

整理上述列举的全部情况，$m$ 个元素的可能情况有如下 $4$ 种。

1. 有 $1$ 个不同元素，出现 $m$ 次。例如 $[1,1,1,1,1,1,1,1,1]$。

2. 有 $m$ 个不同元素，各出现 $1$ 次。例如 $[1,2,3,4,5,6,7,8,9]$。

3. 有 $1$ 个元素出现 $y+1$ 次，其余 $x-1$ 个元素各出现 $y$ 次。例如 $[5,5,6,6,7,7,8,8,8]$。

4. 有 $1$ 个元素出现 $1$ 次，其余 $x$ 个元素各出现 $y$ 次。例如 $[1,1,2,2,3,3,4,4,8]$。

当 $m>1$ 时，上述列举的情况都适用。当 $m=1$ 时，符合情况 1 和情况 2。因此对于任意长度的前缀，都可以根据上述 $4$ 种情况判断前缀是否符合要求。

根据上述分析可知，判断一个前缀是否合法时，需要计算前缀中的不同元素个数、每个元素的出现频率以及每个频率的出现次数，因此需要使用两个哈希表分别记录每个元素的出现频率和每个频率的出现次数，这两个哈希表分别为元素频率哈希表和频率次数哈希表。

从左到右遍历数组 $\text{nums}$，对于 $0\le i<n$，当遍历到下标 $i$ 时，前缀的长度是 $i+1$，令 $\text{num}=\text{nums}[i]$，则 $\text{num}$ 的频率加 $1$。由于在遍历过程中，元素频率只会增加，不会减少，因此需要维护最大频率 $\text{maxFreq}$，即所有元素的出现频率中的最大值，便于判断前缀是否符合要求。具体操作如下。

1. 从元素频率哈希表中得到 $\text{num}$ 的频率 $\text{prevFreq}$（如果 $\text{num}$ 不在元素频率哈希表中则 $\text{prevFreq}=0$），令 $\text{freq}=\text{prevFreq}+1$，则 $\text{prevFreq}$ 和 $\text{freq}$ 分别是 $\text{num}$ 之前的频率和最新的频率。

2. 在元素频率哈希表中将 $\text{num}$ 的频率更新为 $\text{freq}$，并用 $\text{freq}$ 更新 $\text{maxFreq}$。

3. 在频率次数哈希表中将 $\text{prevFreq}$ 的次数减 $1$，将 $\text{freq}$ 的次数加 $1$。如果 $\text{prevFreq}=0$，则不更新 $\text{prevFreq}$ 的次数。

4. 更新两个哈希表之后，元素频率哈希表中的元素（关键字）个数即为不同元素个数，根据不同元素个数和频率次数哈希表判断当前的前缀是否符合要求，即是否符合上述 $4$ 种情况中的至少 $1$ 种。

   1. 如果不同元素个数等于 $1$，则符合情况 1。

   2. 如果不同元素个数等于 $i+1$，则符合情况 2。

   3. 如果频率次数哈希表中，频率 $\text{maxFreq}$ 的次数等于 $1$ 且频率 $\text{maxFreq}-1$ 的次数等于不同元素个数减 $1$，则符合情况 3。

   4. 如果频率次数哈希表中，频率 $\text{maxFreq}$ 的次数等于不同元素个数减 $1$ 且频率 $1$ 的次数等于 $1$，则符合情况 4。

5. 如果当前的前缀符合要求，则将最长前缀的长度更新为 $i+1$（因为当前的前缀长度一定大于之前遍历到的任意前缀的长度），否则不更新最长前缀的长度。

上述操作中，对于每个下标 $i$，更新哈希表和判断当前的前缀是否符合要求的时间都是 $O(1)$，因此总时间复杂度是 $O(n)$。

代码

class Solution {
    public int maxEqualFreq(int[] nums) {
        Map<Integer, Integer> numFreqMap = new HashMap<Integer, Integer>();
        Map<Integer, Integer> freqCountMap = new HashMap<Integer, Integer>();
        int maxPrefixLength = 0;
        int maxFreq = 0;
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            int num = nums[i];
            int prevFreq = numFreqMap.getOrDefault(num, 0);
            int freq = prevFreq + 1;
            numFreqMap.put(num, freq);
            maxFreq = Math.max(maxFreq, freq);
            if (prevFreq > 0) {
                freqCountMap.put(prevFreq, freqCountMap.get(prevFreq) - 1);
                if (freqCountMap.get(prevFreq) == 0) {
                    freqCountMap.remove(prevFreq);
                }
            }
            freqCountMap.put(freq, freqCountMap.getOrDefault(freq, 0) + 1);
            boolean flag = false;
            int distinctNums = numFreqMap.size();
            if (distinctNums == 1) {
                flag = true;
            } else if (distinctNums == i + 1) {
                flag = true;
            } else if (freqCountMap.get(maxFreq) == 1 && freqCountMap.getOrDefault(maxFreq - 1, 0) == distinctNums - 1) {
                flag = true;
            } else if (freqCountMap.get(maxFreq) == distinctNums - 1 && freqCountMap.getOrDefault(1, 0) == 1) {
                flag = true;
            }
            if (flag) {
                maxPrefixLength = i + 1;
            }
        }
        return maxPrefixLength;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{nums}$ 的长度。需要遍历数组 $\text{nums}$，对于每个元素，更新哈希表和判断当前的前缀是否符合要求的时间都是 $O(1)$，因此总时间复杂度是 $O(n)$。

• 空间复杂度：$O(n)$，其中 $n$ 是数组 $\text{nums}$ 的长度。需要使用两个哈希表分别记录每个元素的出现频率和每个频率的出现次数，两个哈希表的空间复杂度都是 $O(n)$。