【算法】js求一个数组的幂集

生成数组所有子集

原创已于 2023-03-15 23:07:22 修改 · 293 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#javascript #前端

于 2022-08-25 00:24:29 首次发布

刷题专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种使用二进制表示来生成一个数组所有可能子集的方法。通过遍历从0到2^n-1的整数，利用每个整数的二进制形式决定原数组哪些元素被选中形成子集。

如题：
例如数组[1，2，3],我们要把它生成[[],[1],[2],[3],[1,2],[1,3],[2,3],[1,2,3]]

const arr = [1,2,3]
const newArr = []
const powerSet = []
// 在[0,2^(n)-1]的整数区间上任取一个值x，x的二进制表示可以用来表示s的一个子集
for(let i = 0;i<Math.pow(2,arr.length);i++) {
  const newNum = i.toString(2).padStart(3,0).split('')
  newArr.push(newNum)
}
// console.log(newArr)
// 对于x的第i位，如果为1，则此子集包含s的第i个元素，否则不包含
for(let j = 0;j<newArr.length;j++) {
  for(let k = 0;k<3;k++) {
    // 如果是1，就存入数组，如果是0，就存入空值
   if(newArr[j][k] == 1) {
    powerSet.push(arr[k])
   } else {
    powerSet.push('')
   }
  }
}
const bwPowerSet = []
// 将数组每3项存到一个数组中，并去除空元素
for(let o = 0;o<powerSet.length;o+= 3) {
const r = powerSet.slice(o,o+3).filter(function (s) {
  return s
})
// 将这些数组push到bwPowerSet数组中，就是要求的子集集合
bwPowerSet.push(r)
}
console.log(bwPowerSet)