1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// @Title: 前 K 个高频元素 (Top K Frequent Elements)
// @Author: 15816537946@163.com
// @Date: 2019-11-17 10:52:24
// @Runtime: 16 ms
// @Memory: 5.4 MB

/*
 * @lc app=leetcode.cn id=347 lang=golang
 *
 * [347] 前 K 个高频元素
 *
 * https://leetcode-cn.com/problems/top-k-frequent-elements/description/
 *
 * algorithms
 * Medium (58.00%)
 * Likes:    154
 * Dislikes: 0
 * Total Accepted:    18.8K
 * Total Submissions: 32.5K
 * Testcase Example:  '[1,1,1,2,2,3]\n2'
 *
 * 给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
 *
 * 示例 1:
 *
 * 输入: nums = [1,1,1,2,2,3], k = 2
 * 输出: [1,2]
 *
 *
 * 示例 2:
 *
 * 输入: nums = [1], k = 1
 * 输出: [1]
 *
 * 说明:
 *
 *
 * 你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
 * 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
 *
 *
 */
 /*
func topKFrequent(nums []int, k int) []int {
	ret := make(map[int]int)
	for _, v := range nums {
		ret[v]++
	}

	// !!!sorted使用二源数组时有道理的,出现相同频率的数字可能有多个
	sorted := make([][]int, len(nums)+1)
	for v, c := range ret {
		sorted[c] = append(sorted[c], v)
	}

	result := make([]int, 0)
	for i := len(sorted) - 1; i > 0 && len(result) < k; i-- {
		if sorted[i] != nil {
			result = append(result, sorted[i]...)
		}
	}
	return result

}	
*/
func topKFrequent(nums []int, k int) []int {
	record := make(map[int]int)
	for _, v := range nums {
		record[v]++
	}

	ret := make([][]int, len(nums)+1)
	for k, v := range record {
		ret[v] = append(ret[v], k)
	}

	res := make([]int, 0)
	for i := len(ret) - 1; i > 0 && len(res) < k; i-- {
		if ret[i] != nil {
			res = append(res, ret[i]...)
		}
	}
	return res
}