通过这道题,我学会了Go的闭包

通过这道题,我学会了Go闭包

先看这道题吧:

给你一个整数数组 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 ,所以它们之间按照数值本身降序排序。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-array-by-increasing-frequency
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

这是leetcode上的一道简单题,处理这道题,常规的思路就是要统计每个数的频率,然后排序频率,再对应道数组里的数,对数组里的数排一遍,这个过程看起来十分麻烦,但是利用闭包可以轻松解决。

什么是闭包呢?

Go中闭包就是引用了自由变量的函数,我们用一个map统计各个数的频率,然后在排序中,使用一个捕获了这个map的闭包,利用map里统计的频率进行排序

func frequencySort(nums []int) []int {
	mymap:=make(map[int]int)
	for _,v:=range nums{
		mymap[v]++
	}
	sort.Slice(nums, func(i, j int) bool {         //mymap是函数外的自由变量,在这里被闭包捕获了
		if mymap[nums[i]]!=mymap[nums[j]]{
			return mymap[nums[i]]<mymap[nums[j]]
		}else{
			return nums[i]>nums[j]
		}
	})
	return nums
}

学会了这个,我感觉自己能把排序玩出花来了。

posted @ 2022-01-02 16:16  博客是个啥?  阅读(36)  评论(0编辑  收藏  举报