2044. 统计按位或能得到最大值的子集数目(2022-3-15)
给你一个整数数组 nums ,请你找出 nums 子集 按位或 可能得到的 最大值 ,并返回按位或能得到最大值的 不同非空子集的数目 。
如果数组 a 可以由数组 b 删除一些元素(或不删除)得到,则认为数组 a 是数组 b 的一个 子集 。如果选中的元素下标位置不一样,则认为两个子集 不同 。
对数组 a 执行 按位或 ,结果等于 a[0] OR a[1] OR … OR a[a.length - 1](下标从 0 开始)。
示例 1:
输入:nums = [3,1]
输出:2
解释:子集按位或能得到的最大值是 3 。有 2 个子集按位或可以得到 3 :
- [3]
- [3,1]
示例 2:
输入:nums = [2,2,2]
输出:7
解释:[2,2,2] 的所有非空子集的按位或都可以得到 2 。总共有 23 - 1 = 7 个子集。
示例 3:
输入:nums = [3,2,1,5]
输出:6
解释:子集按位或可能的最大值是 7 。有 6 个子集按位或可以得到 7 :
- [3,5]
- [3,1,5]
- [3,2,5]
- [3,2,1,5]
- [2,5]
- [2,1,5]
提示:
- 1 <= nums.length <= 16
- 1 <= nums[i] <= 105
解题思路
很容易得出:不停的「按位或」所得出的最大值肯定与当前数组中最大值的二进制长度有关,如果当前数组最大值为5,那么其二进制101
长度为3,按位或的最大值也就是111
,7。
或者换一种说法,只要将数组内的所有元素都按位或一遍,就可以得出最大值。
接下来就是如何计算子集的个数(其实这部分才是关键)
- 一种是状态压缩,将长度最多为16的数组变为1或0的状态数组,枚举所有可能性,并计算是否满足要求
- 另一种是dfs,将选择看作1,0的二叉树,然后进行深度遍历找出满足要求的所有路径
本文选择的是dfs,相差不大本质上都是遍历每一种可能性。代码如下:
var countMaxOrSubsets = function(nums) {
let max = nums.reduce((a,b)=> a | b)
let ret = 0
function dfs(i,v){
if(v == max) ret++
for(let j = i;j < nums.length; j++){
dfs(j + 1,v | nums[j]);
}
}
dfs(0,0)
return ret
};