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
}
|