侧边栏壁纸
博主头像
GabrielxD

列車は必ず次の駅へ。では舞台は?私たちは?

  • 累计撰写 471 篇文章
  • 累计创建 108 个标签
  • 累计收到 9 条评论

目 录CONTENT

文章目录

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

GabrielxD
2022-09-19 / 0 评论 / 0 点赞 / 15 阅读 / 526 字 / 正在检测是否收录...

题目

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;
    }
}
0

评论区