sorted的key的一个比较取巧的用法

题目

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

我的答案:先哈希+再排序

class Solution:
    def frequencySort(self, nums: List[int]) -> List[int]:
        if len(nums) <= 1:
            return nums

        di,redi,tar = {},{},[]

        for i in nums:
            di[i] = di.get(i,0) + 1

        for i in nums:
            value = di.get(i)
            if value in redi:
                redi[value].append(i)
                redi[value] = sorted(redi[value],reverse = True)
            else:
                redi[value] = [i]
        
        n = len(nums)

        for i in range(n+1):
            if i in redi:
                tar.extend(redi.get(i))

        return tar


本来的思路是哈希完先按照元素的数量增序排序,再按照 相同数量的元素大小逆序排序,但是这个key不会写,官方答案就是这个思路。
代码如下:

class Solution:
    def frequencySort(self, nums: List[int]) -> List[int]:
        if len(nums) <= 1:
            return nums

        di = {}

        for i in nums:
            di[i] = di.get(i,0) + 1


        return sorted(nums,key = lambda x:(di.get(x),-x))
      

key是一个元组,元组两个元素,先按照元素的数量增序排列,再按照元素的负数增序排列(可不就是降序吗!)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值