Top K Frequent Elements - LeetCode 347

top_k_frequent_elements.go

func topKFrequent(nums []int, k int) []int {
    freqMap := make(map[int]int)
    for _, n := range nums {
        freqMap[n]++
    }
    
    freqArr := make([][]int, len(nums)+1)
    for k, v := range freqMap {
        freqArr[v] = append(freqArr[v], k)
    }
    
    var result []int
    freqArrLen := len(freqArr)
    
    for i := freqArrLen-1; i > -1 && k > 0; i-- {
        for _, n := range freqArr[i] {
            result = append(result, n)
            k--
            if k == 0 {
                return result
            }
        }    
    }
    
    return result
}