LeetCode.H1224.最大相等频率

LeetCode.H1224

题目：

题目大意：

​ 如图所示。

数据范围：

如图所示

思路：

​ 从前向后遍历，边遍历边维护:

• cntmap:记录当前 各个数字出现的次数。
• fremap:记录当前 各个出现次数有几个数字。
• maxcnt:记录当前 出现的最大次数。
• 例：nums = [a,a,a,b,b,c,c]，遍历到最后一个元素c时:
  • cntmap[a] = 3, cntmap[b] = 2, cntmap[c] = 2
  • fremap[3] = 1 (出现次数为1的只有a一个), fremap[2] = 2（出现次数为2的有b、c两个）

那么当满足题意时，就更新ans，存在以下三种情况：（假设当前遍历到第i（从0开始）个元素，共有nowlen = i + 1 个数）

• 当 maxcnt == 1时，说明当前所有元素出现次数都为1，这时候随便删除一个元素，剩下的元素出现个数依旧相等（都为1）。
• 当 maxcnt * fremap[maxcnt] + 1 == nowlen时，说明当前是一个出现次数为1的元素，其余各个元素出现次数都相等（都为maxcnt）的前缀。只要把这一个出现次数为1的元素删去，剩下的各个元素的出现次数就都相等了。
• 当 (maxcnt - 1) * fremap[maxcnt - 1] + maxcnt == nowlen时，说明当前是一个出现次数为maxcnt的元素，其余各个元素出现次数都相等（都为maxcnt - 1）的前缀，那么只需要把这个出现次数为maxcnt的元素删掉一个，使其出现次数变为maxcnt - 1，那么所有的元素的出现次数就都相等了。

代码：

import java.util.HashMap;
import java.util.Map;

class Solution {
    public int maxEqualFreq(int[] nums) {
        //记录当前 各个数字出现的次数。
        Map<Integer, Integer> cntmap = new HashMap<>();
        //记录当前 各个出现次数有几个数字。
        Map<Integer, Integer> fremap = new HashMap<>();
        //记录当前 出现的最大次数。
        int maxcnt = -1, ans = -1;
        for (int i = 0; i < nums.length; i ++ ){
            //当前前缀的长度
            int nowlen = i + 1;
            //更新数字的出现次数
            cntmap.put(nums[i], cntmap.getOrDefault(nums[i], 0) + 1);
            int cntnums = cntmap.get(nums[i]);
            //将更新完后的数字出现次数与 最大次数比较。
            maxcnt = Math.max(maxcnt, cntnums);
            //更新 各个这个出现次数有几个数字。
            fremap.put(cntnums, fremap.getOrDefault(cntnums, 0) + 1);
            //注意这个减操作
            /*
            假设当前为aabbcc，则fremap[2] = 3，
            表示出现次数为2的数字有三个（a、b、c）
            那么新加入一个c后变为aabbccc，c的出现次数变为3
            那么fremap[3] = 1， fre[2] = 3 - 1 = 2 
            (减1是因为c的出现次数变为3了，只剩下a、b这两个出现次数为2的数字)
             */
            fremap.put(cntnums - 1, fremap.getOrDefault(cntnums - 1, 0) - 1);
            if (maxcnt == 1){
                ans = nowlen;
            }
            if (maxcnt * fremap.get(maxcnt) + 1 == nowlen){
                ans = nowlen;
            }
            if ((maxcnt - 1) * fremap.get(maxcnt - 1) + maxcnt == nowlen){
                ans = nowlen;
            }
        }
        return ans;
    }
}

public class Main {
    public static void main(String[] args) {
        Solution solution = new Solution();
//        int[] nums = new int[]{2,2,1,1,5,3,3,5};
//        int[] nums = new int[]{1,1,1,2,2,2,3,3,3,4,4,4,5};
        int[] nums = new int[]{1,1,1,2,2,2};
        System.out.println(solution.maxEqualFreq(nums));
    }

}

时空复杂度分析等：

• 时间复杂度 : O(n)

• 空间复杂度 : O(n)

题目链接：

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