【Leetcode 1636 】 按照频率将数组升序排序—— 比官方还简洁,时间击败 100%,空间击败87.5

54 阅读1分钟

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

比官方还简洁,时间击败 100%,空间击败87.5

function frequencySort(nums: number[]): number[] {
  const numMap = new Map<number, number>();
  //计算次数
  for (const n of nums) {
    numMap.set(n, (numMap.get(n) || 0) + 1);
  }
  nums.sort((a, b) => {
    const countA = numMap.get(a)!,
      countB = numMap.get(b)!;
      // countA - countB === 0 证明  countA === countB
      // countA === countB ?  b - a ?  countA - countB 的简写
    return countA - countB || b - a;
  });
  return nums;
}