【排序, 哈希表】按照频率将数组升序排序

题目

1636. 按照频率将数组升序排序

---

给你一个整数数组 nums ，请你将数组按照每个值的频率 升序 排序。如果有多个值的频率相同，请你按照数值本身将它们 降序 排序。

请你返回排序后的数组。

示例 1：

输入：nums = [1,1,2,2,2,3]
输出：[3,1,1,2,2,2]
解释：'3' 频率为 1，'1' 频率为 2，'2' 频率为 3 。

示例 2：

输入：nums = [2,3,1,3,2]
输出：[1,3,3,2,2]
解释：'2' 和 '3' 频率都为 2 ，所以它们之间按照数值本身降序排序。

示例 3：

输入：nums = [-1,1,-6,4,5,-6,1,4,1]
输出：[5,-1,4,4,-6,-6,1,1,1]

提示：

• 1 <= nums.length <= 100
• -100 <= nums[i] <= 100

解题

方法一：自定义排序 哈希表

思路

先统计出每个数出现的次数，再按照题目要求自定义排序即可。

代码

class Solution {
    public int[] frequencySort(int[] nums) {
        int[] counts = new int[201];
        List<Integer> list = new ArrayList<Integer>();
        for (int num : nums) {
            ++counts[num + 100];
            list.add(num);
        }
        list.sort((a, b) -> counts[a + 100] == counts[b + 100] ?
            b - a : counts[a + 100] - counts[b + 100]);
        int n = nums.length;
        int[] sorted = new int[n];
        for (int i = 0; i < n; ++i) sorted[i] = list.get(i);
        return sorted;
    }
}

class Solution {
public:
    vector<int> frequencySort(vector<int>& nums) {
        int counts[201] = {0};
        for (int& num : nums) ++counts[num + 100];
        sort(nums.begin(), nums.end(), [&counts](int& a, int& b) {
            return counts[a + 100] == counts[b + 100] ?
                a > b : counts[a + 100] < counts[b + 100];
        });
        return nums;
    }
};

不使用自定义排序而是借助树集的做法：

class Solution {
    public int[] frequencySort(int[] nums) {
        Map<Integer, Integer> counts = new HashMap<>();
        for (int num : nums) counts.put(num, counts.getOrDefault(num, 0) + 1);
        Map<Integer, Set<Integer>> groups = new TreeMap<>();
        counts.forEach((k, v) -> {
            groups.putIfAbsent(v, new TreeSet<>(Comparator.reverseOrder()));
            groups.get(v).add(k);
        });
        int n = nums.length;
        int[] sorted = new int[n];
        int i = 0;
        for (var entry : groups.entrySet()) {
            int cnt = entry.getKey();
            for (int num : entry.getValue()) {
                for (int j = 0; j < cnt; ++j) sorted[i++] = num;
            }
        }
        return sorted;
    }
}