如何不转换为数组即可从JavaScript Set中获取随机元素

可以直接实现，不需要将Set转为数组，核心利用Set本身的迭代器特性完成随机位置元素的读取。

实现思路 #

• 通过Set.prototype.size属性O(1)获取Set的元素总数
• 生成0到元素总数-1区间内的随机整数作为目标下标
• 调用Set.prototype.values()获取Set的迭代器实例
• 多次调用迭代器的next()方法遍历到目标下标位置，返回对应元素即可

代码实现 #

function getRandomSetItem(set) {
  const size = set.size;
  // 空Set直接返回undefined
  if (size === 0) return undefined;
  // 生成合法随机下标
  const randomIndex = Math.floor(Math.random() * size);
  const iterator = set.values();
  let currentItem = iterator.next();
  // 迭代到目标位置
  for (let i = 0; i < randomIndex; i++) {
    currentItem = iterator.next();
  }
  return currentItem.value;
}

// 针对你的示例测试
const mySet = new Set()
mySet.add(1)
mySet.add(2)
mySet.add(3)
console.log(getRandomSetItem(mySet)) // 随机输出1、2、3中的一个

方案优劣势说明 #

• 优势：无需额外开辟数组内存存储Set元素，适合Set体量极大、仅需单次或少量获取随机元素的场景
• 劣势：单次获取元素的时间复杂度为O(n)，如果需要频繁获取随机元素，预先将Set转数组后随机访问（单次O(1)时间复杂度）的整体性能更优

内容的提问来源于stack exchange，提问作者Caleb Oki