2021-11-12:前 K 个高频元素。给你一个整数数组 nums 和
2023-03-14 22:45:32 时间
2021-11-12:前 K 个高频元素。给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。提示:1 <= nums.length <= 105,k 的取值范围是 1, 数组中不相同的元素的个数,题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。进阶:你所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n 是数组大小。力扣347。
答案2021-11-12:
门槛堆。小根堆。
代码用golang编写。代码如下:
package main
import (
"fmt"
"sort"
)
func main() {
nums := []int{1, 1, 1, 2, 2, 3}
k := 2
ret := topKFrequent(nums, k)
fmt.Println(ret)
}
type Node struct {
num int
count int
}
func NewNode(k int) *Node {
res := &Node{}
res.num = k
res.count = 1
return res
}
func topKFrequent(nums []int, k int) []int {
map0 := make(map[int]*Node)
for _, num := range nums {
if _, ok := map0[num]; !ok {
map0[num] = NewNode(num)
} else {
map0[num].count++
}
}
heap := make([]*Node, 0)
for _, node := range map0 {
if len(heap) < k || (len(heap) == k && node.count > heap[0].count) {
//heap.add(node);
heap = append(heap, node)
sort.Slice(heap, func(i, j int) bool {
return heap[i].count > heap[j].count
})
}
if len(heap) > k {
heap = heap[1:]
}
}
ans := make([]int, k)
index := 0
for len(heap) > 0 {
ans[index] = heap[0].num
heap = heap[1:]
index++
}
return ans
}
执行结果如下:
相关文章
- 在 Go 里用 CGO?这 7 个问题你要关注!
- 9款优秀的去中心化通讯软件 Matrix 的客户端
- 求职数据分析,项目经验该怎么写
- 在OKR中,我看到了数据驱动业务的未来
- 火山引擎云原生大数据在金融行业的实践
- OpenHarmony富设备移植指南(二)—从postmarketOS获取移植资源
- 《数据成熟度指数》报告:64%的企业领袖认为大多数员工“不懂数据”
- OpenHarmony 小型系统兼容性测试指南
- 肯睿中国(Cloudera):2023年企业数字战略三大趋势预测
- 适用于 Linux 的十大命令行游戏
- GNOME 截图工具的新旧截图方式
- System76 即将推出的 COSMIC 桌面正在酝酿大变化
- 2GB 内存 8GB 存储即可流畅运行,Windows 11 极致精简版系统 Tiny11 发布
- 迎接 ecode:一个即将推出的具有全新图形用户界面框架的现代、轻量级代码编辑器
- loongarch架构介绍(三)—地址翻译
- Go 语言怎么解决编译器错误“err is shadowed during return”?
- 敏捷:可能被开发人员遗忘的部分
- Denodo预测2023年数据管理和分析的未来
- 利用数据推动可持续发展
- 在 Vue3 中实现 React 原生 Hooks(useState、useEffect),深入理解 React Hooks 的