【Leetcode】1224. Maximum Equal Frequency

题目地址：

https://leetcode.com/problems/maximum-equal-frequency/description/

给定一个数组$A$，问其最长前缀，使得这个前缀删去一个数后剩下的数的频率都相等。只需要返回这个前缀长度。

直接从小到大枚举前缀，开两个哈希表，一个是记录每个数字出现了多少次，另一个是记录各个频率出现了多少次。考虑如何判断当前前缀是否符合条件。如果频率的种类超过$2$个那肯定不行；如果频率种类为$2$个，那么要么某一个数频率为$1$并且频率$1$的数也只有一个，则将这个只出现$1$次的数删去即可，要么某个数的频率为$1$，并且其余数都等于这个数减$1$；如果频率种类为$1$个，那么要么频率为$1$，要么这个频率出现的次数为$1$。代码如下：

class Solution {
 public:
#define x first
#define y second
  int maxEqualFreq(vector<int>& nums) {
    unordered_map<int, int> cnt, mp;
    int res = 0;
    for (int i = 0; i < nums.size(); i++) {
      int x = nums[i];
      if (cnt.count(x)) {
        mp[cnt[x]]--;
        if (!mp[cnt[x]]) mp.erase(cnt[x]);
      }

      cnt[x]++;
      mp[cnt[x]]++;

      if (mp.size() == 1) {
        for (auto& [k, v] : mp)
          if (k == 1 || v == 1) res = i + 1;
      } else if (mp.size() == 2) {
        vector<pair<int, int>> v(mp.begin(), mp.end());
        sort(v.begin(), v.end());
        if (v[0].x == 1 && v[0].y == 1 || v[0].x + 1 == v[1].x && v[1].y == 1)
          res = i + 1;
      }
    }

    return res;
  }
};

时空复杂度$O(n)$。