通过这道题,我学会了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
}
学会了这个,我感觉自己能把排序玩出花来了。