LeetCode1224. 最大相等频率【思维】

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

示例 3：

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

示例 4：

输入：nums = [10,2,8,9,3,8,1,5,2,3,7,6]
输出：8

对成立的情况进行分类讨论，一共四种，然后进行模拟。

提示：

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

class Solution {
    //const int maxn = 1e5 + 10;
    int a[int(1e5 + 10)] = {0};
    int b[int(1e5 + 10)] = {0};
public:
    int maxEqualFreq(vector<int>& nums) {
        int max_time = 0, ans = 0, dif = 0;
        for(int i = 0; i < nums.size(); i++) {
            int x = nums[i];
            if(a[x]++ == 0) dif++;
            b[a[x]]++;
            if(b[a[x] - 1] > 0) b[a[x] - 1]--;
            max_time = max(max_time, a[x]);
            if(max_time == i + 1 || b[1] == i + 1 || (b[max_time] == 1 && b[max_time - 1] == dif - 1) || (b[1] == 1 && b[max_time] == dif - 1)) {
                ans = i + 1;
            }
        }
        return ans;
    }
};