学习目标:
- 前 K 个高频元素
leetcode原题链接
学习内容:
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
提示:
- 1 <= nums.length <= 10^5
- k 的取值范围是 [1, 数组中不相同的元素的个数]
- 题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的
解题思路:
思路一 通过Set和Map
利用Set不包含重复项的特性,先设置一个去重后的数组arr,再创建一个map作为映射,遍历给定的数字数组nums,如果此时遍历的值不存在之前创建的map里,则将它加入到映射中,值为1,如果已经存在于映射中,则将之前映射的对应值加1。最后再通过sort方法对最开始创建的去重数组arr进行从大到小排序,排序方法就是映射中各个值的大小比较,最后,返回排序后的前k项即可得到最终结果。
实现代码:
/**
* @param {number[]} nums
* @param {number} k
* @return {number[]}
*/
var topKFrequent = function (nums, k) {
const arr = [...new Set(nums)]
const map = new Map()
nums.map(item => {
if (map.has(item)) {
map.set(item, map.get(item) + 1)
} else {
map.set(item, 1)
}
})
return arr.sort((a, b) => map.get(b) - map.get(a)).slice(0, k)
}
思路二 桶排序
实现方法就是先将数据划分到有限个桶里,再对桶进行排序,这里还是用map存储频次,用数组表达桶,频次作为数据下标,针对不同的频次的数字,聚合。
实现代码:
var topKFrequent = function (nums, k) {
let map = new Map()
let arr = [...new Set(nums)]
nums.map(item => {
if (map.has(item)) {
map.set(item, map.get(item) + 1)
} else {
map.set(item, 1)
}
})
if (map.size < k) {
return [...map.keys()]
}
// 桶排序
return bucket(map, k)
}
function bucket(map, k) {
let res = []
let arr = []
map.forEach((value, key) => {
// 可以利用频次作为下标,将数据分配到桶里
if (!arr[value]) {
arr[value]=[key]
} else {
arr[value].push(key)
}
})
for (let i = arr.length - 1; i >= 0 && res.length < k; i--){
if (arr[i]) {
res.push(...arr[i])
}
}
return res
}