关于golang:Golang力扣Leetcode中级算法排序和搜索前K个高频元素哈希表存储

42次阅读

共计 550 个字符，预计需要花费 2 分钟才能阅读完成。

题目：给你一个整数数组 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))
}

提交截图：

正文完