最大相等频率

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第24天，点击查看活动详情

1224. 最大相等频率 - 力扣（LeetCode）

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

解题

/**
 * @param {number[]} nums
 * @return {number}
 */
var maxEqualFreq = function (nums) {
  const counts = new Map();
  const freqs = new Map();

  const valid = () => {
    if (freqs.size > 2) return false;
    let kvs = Array.from(freqs);
    if (kvs.length === 1) {
      return kvs[0][0] === 1 || kvs[0][1] === 1;
    }
    kvs.sort((a, b) => a[0] - b[0]);
    if (kvs[0][0] === 1 && kvs[0][1] === 1) {
      return true;
    }
    kvs.sort((a, b) => a[1] - b[1] || b[0] - a[0]);
    if (kvs[0][1] === 1 && kvs[0][0] - kvs[1][0] === 1) {
      return true;
    }
    return false;
  };
  let res = 0;
  for (let i = 0; i < nums.length; i++) {
    let count = counts.get(nums[i]) || 0;
    if (count) {
      if (freqs.get(count) === 1) {
        freqs.delete(count);
      } else {
        freqs.set(count, freqs.get(count) - 1);
      }
    }
    count++;
    counts.set(nums[i], count);
    freqs.set(count, (freqs.get(count) || 0) + 1);
    if (i < 3 || valid(freqs)) {
      res = i + 1;
    }
  }
  return res;
};