巧妙利用位运算——leetcode 78子集

看到题目,总觉得自己会写,但就是写不出来。原题目链接:leetcode 78 子集


给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:
输入: nums = [1,2,3]
输出:
	[
	  [3],
	  [1],
	  [2],
	  [1,2,3],
	  [1,3],
	  [2,3],
	  [1,2],
	  []
	]

题目不难理解,我想到的是

0个元素 得 []1个元素 得  [3], [1], [2],2个元素得  [1,3],  [2,3],  [1,2],3个元素 得 [1,2,3]

代码可以实现,但肯定不优雅,看了官方题解,给出的真是优雅,就是不好想。
比如说原来数组是{5, 2,9},每个元素要么被选,要么不被选,选择记为1,不选记为0
在这里插入图片描述
如图所示:
第一种都不选 就是 000
第二种,只选最后一个 就是 001
……
把0 1看做是二进制里的0 和1,n 个元素,那子集的总数是 2n个,拿到其中一个二进制,比如是010,那对应的子集就是 {2},直接看代码 ,更容易理解

    public List<List<Integer>> subsets(int[] nums) {
        int n = nums.length;
        List<List<Integer>> result = new ArrayList<List<Integer>>(); // 记录结果
        List<Integer> temp = new ArrayList<Integer>();
        // 子集总个数为2的n次方,也就是 1 << n,外层循环中的bit,就是上图里的1 2 3 4 5 6 7
        for(int bit = 0; bit < 1 << n; bit++){
            temp.clear(); // 清空集合
            for(int i = 0; i < n; i++){ // 循环看 bit 的每一位
                if((bit & (1 << i)) != 0){ // bit 的某一位是不是1,若是就取出该元素
                    temp.add(nums[i]);
                }
            }
            result.add(new ArrayList<Integer>(temp));
        }
        return result;
    }

这就是套路,位运算用的十分的巧妙,确实我想不到。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值