[Golang]力扣Leetcode—中级算法—排序和搜索—前K个高频元素（哈希表存储）

[Golang]力扣Leetcode—中级算法—排序和搜索—前K个高频元素（哈希表存储）

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

提交截图：