Top K Frequent Elements

Top K Frequent Elements

要求

Given a non-empty array of integers, return the k most frequent elements.

Example 1:
Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]
Example 2:
Input: nums = [1], k = 1
Output: [1]

思路

top k的问题虽然没有接触过，但是最简单的就是排序后从后往前拿出来k项，这就没意思了。
我的思路：利用堆的特性来pop堆顶元素，进行k次。 
网上思路：快排，桶排序。
我的理解：
	1.快排的思想来解决，想了想也是可以，不断的划分，不需要排序完，只需要满足找到
top k就可以
	2.桶排序思想的，说实话下午没仔细看，我没想到是根据什么映射把桶的含义作为次数
的。之后看明白了再来补一篇。
	3.我这里用了堆，我用堆的原因是因为它最直观，最容易理解。

代码

type pair struct {
	key int
	val int
}

type pairs []pair

func (p pairs) Len() int {
	return len(p)
}

func (p pairs) Swap(i, j int) {
	p[i], p[j] = p[j], p[i]
}

//默认的时间里小跟堆，这里吧小跟堆的比较反一下就成了大根堆
//同时这里是按照val类型当做排序关键字来进行排序的
func (p pairs) Less(i, j int) bool {
	return p[i].val >= p[j].val
}

func (p *pairs) Push(x interface{}) {
	*p = append(*p, x.(pair))
}
func (p *pairs) Pop() interface{} {
	old := *p
	n := len(old)
	x := old[n-1]
	*p = old[0 : n-1]
	return x
}

func topKFrequent(nums []int, k int) []int {
	tem := make(map[int]int, 20)

	//数字作为键，数字出现的次数作为值记录到哈希表中
	for _, num := range nums {
		_, ok := tem[num]
		if ok {
			tem[num]++
		} else {
			tem[num] = 1
		}
	}

	//遍历哈希表，把哈希表的元素按照val类型建立大根堆
	var h pairs
	for k, v := range tem {
		h = append(h, pair{k, v})
	}

	heap.Init(&h)

	//根据题意，筛选出符合的K项
	var ret []int
	for {
		x := heap.Pop(&h)
		ret = append(ret, (x.(pair).key))

		k--
		if k == 0 {
			break
		}
	}

	return ret
}

关于作者

大四学生一枚，分析数据结构，面试题，golang，C语言等知识。QQ交流群：521625004。微信公众号：后台技术栈。