请求解释Top K Frequent Elements算法中第4行代码的含义
关于Top K Frequent Elements算法中计数逻辑的解析
你在拆解Top K Frequent Elements算法时,对代码中第4行的计数逻辑有疑问,先把完整代码贴出来:
var topKFrequent = (nums, k) => { let map = {} for(let element of nums){ map[element] = (map[element] || 0) + 1 } return Object.entries(map).sort((a,b) => b[1] -a[1]).map(val=>Number(val[0])).slice(0,k); }
核心逻辑解析:map[element] = (map[element] || 0) + 1
这行是JavaScript里统计元素出现次数的简写技巧,完全等价于下面的常规条件判断写法:
if (map[element]) { // 元素已经统计过,计数加1 map[element] += 1; } else { // 第一次遇到该元素,初始化为1 map[element] = 1; }
之所以能这么写,是利用了JavaScript中||(逻辑或)的特性:
- 当
map[element]有值(比如之前已经统计过,值是1、2这类数字),map[element] || 0会直接返回map[element]的当前值 - 当
map[element]不存在(第一次遍历到该元素,对象中无此键,值为undefined),undefined属于逻辑假值,此时map[element] || 0会返回0 - 最后给结果加1,再赋值回
map[element],就完成了"首次出现计数为1,重复出现计数递增"的逻辑
补充:整个函数的流程
- 初始化空对象
map,用来存每个元素的出现次数 - 遍历输入数组
nums,用上面的逻辑统计每个元素的次数 - 把
map转成键值对数组(格式如[[元素1, 次数1], [元素2, 次数2]]) - 按次数从高到低排序
- 提取排序后的元素,转成数字类型,取前
k个结果返回
内容的提问来源于stack exchange,提问作者poopmachine
相关产品推荐
相关产品推荐

