2044. 统计按位或能得到最大值的子集数目(medium)

最新推荐文章于 2022-11-20 15:16:44 发布

weixin_45760415

最新推荐文章于 2022-11-20 15:16:44 发布

阅读量207

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_45760415/article/details/123497751

版权

leetcode 专栏收录该内容

43 篇文章 0 订阅

订阅专栏

统计按位或能得到最大值的子集数目
给你一个整数数组 nums ，请你找出 nums 子集按位或可能得到的最大值，并返回按位或能得到最大值的不同非空子集的数目。
如果数组 a 可以由数组 b 删除一些元素（或不删除）得到，则认为数组 a 是数组 b 的一个子集。如果选中的元素下标位置不一样，则认为两个子集不同。
对数组 a 执行按位或，结果等于 a[0] OR a[1] OR … OR a[a.length - 1]（下标从 0 开始）。

1.回溯
每次回溯时，如果pos==n，判断是否更新结果。否则，枚举有两种情况，有当前数字或无当前数字，所以关于pos+1调用两次回溯函数。

class Solution {
public:
    int maxOR=0;
    int res=0;
    void traceback(vector<int>& nums,int pos,int sum){
        if(pos>=nums.size()){
            if(sum>maxOR){
                maxOR=sum;
                res=1;
            }else if(sum==maxOR){
                res++;
            }
            cout<<sum<<endl;
            return; 
        } 
        traceback(nums,pos+1,sum|nums[pos]);
        traceback(nums,pos+1,sum);
    }
    int countMaxOrSubsets(vector<int>& nums) {
        traceback(nums,0,0);
        return res;
    }
};

2.使用位运算枚举子数组
n位二进制数范围内有2的n次方个数字，与子数组的数目一致。二进制数对应位为1则在子数组中，否则不在。对于二进制数i，要判断第j位是否为1：(i>>j)&1.
求2的n次方：1<<n

class Solution {
public:
    int countMaxOrSubsets(vector<int>& nums) {
        int maxOR=0,res=0;
        int n=1<<nums.size();
        for(int i=0;i<n;i++){
            int sum=0;
            for(int j=0;j<nums.size();j++){
                if(i&(1<<j)){
                    sum|=nums[j];
                }
            }
            if(sum>maxOR){
                maxOR=sum;
                res=1;
            }else if(sum==maxOR){
                res++;
            }
        }  
        return res;      
    }
};