[Leetcode] 347. Top K Frequent Elements

2023-02-24 114 words One minute

Contents

https://leetcode.com/problems/top-k-frequent-elements/description/

문제 자체는 어렵지 않을 텐데 map(hash table)을 value 값으로 정렬하는데 시간을 많이 잡아먹었다.

go에선 sort interface 사용법을 알았다면 금방 풀수 있었을듯

 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


func topKFrequent(nums []int, k int) []int {
    prev := nums[0]

    tmp := map[int]int{}
    tmp[nums[0]] = 1

    for i := 1; i < len(nums); i++ {
        tmp[nums[i]] += 1
        if prev != nums[i] {
            prev = nums[i]
        }
    }
    

    keys := []int{}
    for k, _ := range tmp {
        keys = append(keys, k)
    }

    sort.Slice(keys, func(i, j int) bool {
        return tmp[keys[i]] > tmp[keys[j]]
    })

    return keys[:k]
}