题目:给你一个整数数组 nums 和一个整数 k ,请你返回其中呈现频率前 k 高的元素。你能够按 任意程序 返回答案。
链接: 力扣Leetcode—中级算法—排序和搜寻—前K个高频元素.
示例 1:
输出: nums = [1,1,1,2,2,3], k = 2
输入: [1,2]
示例 2:
输出: nums = [1], k = 1
输入: [1]
标签:数组、哈希表、分治、桶排序、计数、疾速抉择、排序、堆(优先队列)
思路:用哈希表来存储次数,再遍历这个哈希表,返回其中呈现频率前 k 高的元素
全副Go代码如下:
package mainimport ( "fmt")func topKFrequent(nums []int, k int) []int { // 用一个map来存储数字以及呈现的次数 m := make(map[int]int) for _, v := range nums { m[v]++ } fmt.Println(m) // 定义存储最初后果的数组 var res []int for i := 0; i < k; i++ { res = append(res, 0) f := 0 for n, v := range m { if v > f { res[i] = n f = v } } m[res[i]] = -1 } return res}func main() { a := []int{3, 0, 1, 0} fmt.Println(topKFrequent(a, 1))}
提交截图: