2021-11-12:前 K 个高频元素。给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。提示:1 <= nums.length <=

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
}

执行结果如下:
图片


左神java代码

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

福大大架构师每日一题

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值