题目:给你一个整数数组 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 main
import (
"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))
}